Spaces:
Runtime error
Runtime error
Commit
·
2549c67
1
Parent(s):
ac7ad0e
remove new embedding
Browse files
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 |
-
|
532 |
-
|
533 |
-
|
534 |
-
)
|
535 |
|
536 |
-
embeddings =
|
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)
|