OuroborosM commited on
Commit
2549c67
·
1 Parent(s): ac7ad0e

remove new embedding

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -527,13 +527,13 @@ embed_model_id = 'sentence-transformers/all-MiniLM-L6-v2'
527
 
528
  # device = f'cuda:{cuda.current_device()}' if cuda.is_available() else 'cpu'
529
  device = 'cpu'
530
- embed_model = HuggingFaceEmbeddings(
531
- model_name=embed_model_id,
532
- model_kwargs={'device': device},
533
- encode_kwargs={'device': device, 'batch_size': 32}
534
- )
535
 
536
- embeddings = embed_model
537
 
538
 
539
  # embeddings = OpenAIEmbeddings(deployment="model_embedding_2", chunk_size=15)
 
527
 
528
  # device = f'cuda:{cuda.current_device()}' if cuda.is_available() else 'cpu'
529
  device = 'cpu'
530
+ # embed_model = HuggingFaceEmbeddings(
531
+ # model_name=embed_model_id,
532
+ # model_kwargs={'device': device},
533
+ # encode_kwargs={'device': device, 'batch_size': 32}
534
+ # )
535
 
536
+ embeddings = embeddings_openai
537
 
538
 
539
  # embeddings = OpenAIEmbeddings(deployment="model_embedding_2", chunk_size=15)