Spaces:
Sleeping
Sleeping
Update utils.py
Browse files
utils.py
CHANGED
|
@@ -101,7 +101,7 @@ embedding_model = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mp
|
|
| 101 |
|
| 102 |
#Generate embeddings for a given text
|
| 103 |
def get_embeddings(text):
|
| 104 |
-
return embedding_model.embed_query([text]
|
| 105 |
|
| 106 |
|
| 107 |
# Create or connect to a Qdrant collection
|
|
|
|
| 101 |
|
| 102 |
#Generate embeddings for a given text
|
| 103 |
def get_embeddings(text):
|
| 104 |
+
return embedding_model.embed_query([text])[0] #, task_type='search_document'
|
| 105 |
|
| 106 |
|
| 107 |
# Create or connect to a Qdrant collection
|