Spaces:
Sleeping
Sleeping
Goodnight7
commited on
Commit
•
e6d300b
1
Parent(s):
1deade6
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
|