Subha95 commited on
Commit
f277648
·
verified ·
1 Parent(s): b0a64a7

Update chatbot_rag.py

Browse files
Files changed (1) hide show
  1. chatbot_rag.py +5 -2
chatbot_rag.py CHANGED
@@ -15,8 +15,11 @@ def build_qa():
15
  # 1. Embeddings
16
  embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
17
 
18
- # 2. Vector DB (already persisted in Kaggle, same folder name in HF Space)
19
- vectorstore = Chroma(persist_directory="db", embedding_function=embeddings)
 
 
 
20
 
21
  # 3. LLM
22
  model_id = "microsoft/phi-2"
 
15
  # 1. Embeddings
16
  embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
17
 
18
+ vectorstore = Chroma(
19
+ persist_directory="db",
20
+ collection_name="rag-docs",
21
+ embedding_function=embeddings,
22
+ )
23
 
24
  # 3. LLM
25
  model_id = "microsoft/phi-2"