Hugging Face's logo

Spaces:
jonathanjordan21
/
medical_chatbot
Sleeping

App Files Community
jonathanjordan21 commited on
Commit
57ac245
•
1 Parent(s): 9e2621c

Update custom_llm.py

Browse files
Files changed (1) hide show
  1. custom_llm.py +1 -1
custom_llm.py CHANGED
@@ -23,7 +23,7 @@ import datasets
23
 
24
 
25
  def create_vectorstore():
26
- data = load_datasets("ruslanmv/ai-medical-chatbot", split='train')
27
  emb_model = HuggingFaceEmbeddings(model_name='sentence-transformers/paraphrase-multilingual-mpnet-base-v2', encode_kwargs={'normalize_embeddings': True})
28
  faiss = FAISS.from_texts(data['Doctor'], emb_model)
29
  return faiss
 
23
 
24
 
25
  def create_vectorstore():
26
+ data = load_dataset("ruslanmv/ai-medical-chatbot", split='train')
27
  emb_model = HuggingFaceEmbeddings(model_name='sentence-transformers/paraphrase-multilingual-mpnet-base-v2', encode_kwargs={'normalize_embeddings': True})
28
  faiss = FAISS.from_texts(data['Doctor'], emb_model)
29
  return faiss