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)
|