Spaces:
Sleeping
Sleeping
Update chatbot_rag.py
Browse files- 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 |
-
|
| 19 |
-
|
|
|
|
|
|
|
|
|
|
| 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"
|