Goodnight7 commited on
Commit
e6d300b
1 Parent(s): 1deade6

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
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], task_type='search_document')[0]
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