Nada commited on
Commit
1a5a8ad
·
1 Parent(s): 09389ab
Files changed (1) hide show
  1. chatbot.py +2 -4
chatbot.py CHANGED
@@ -294,11 +294,9 @@ Response:"""
294
  )
295
 
296
  # Setup embeddings for vector search
297
- embedding_model = SentenceTransformer(
298
- "sentence-transformers/all-MiniLM-L6-v2",
299
- cache_folder=os.getenv("HF_HOME", "/tmp/huggingface")
300
  )
301
- self.embeddings = HuggingFaceEmbeddings(model=embedding_model)
302
 
303
  # Setup vector database for retrieving relevant past conversations
304
  if therapy_guidelines_path and os.path.exists(therapy_guidelines_path):
 
294
  )
295
 
296
  # Setup embeddings for vector search
297
+ self.embeddings = HuggingFaceEmbeddings(
298
+ model_name="sentence-transformers/all-MiniLM-L6-v2"
 
299
  )
 
300
 
301
  # Setup vector database for retrieving relevant past conversations
302
  if therapy_guidelines_path and os.path.exists(therapy_guidelines_path):