Spaces:
Running
Running
Upload app.py with huggingface_hub
Browse files
app.py
CHANGED
@@ -158,9 +158,11 @@ def expand_query(state):
|
|
158 |
# Initialize the Chroma vector store for retrieving documents
|
159 |
vector_store = Chroma(
|
160 |
collection_name="nutritional_hypotheticals",
|
161 |
-
persist_directory="./
|
|
|
162 |
embedding_function=embedding_model
|
163 |
)
|
|
|
164 |
|
165 |
# Create a retriever from the vector store
|
166 |
retriever = vector_store.as_retriever(
|
|
|
158 |
# Initialize the Chroma vector store for retrieving documents
|
159 |
vector_store = Chroma(
|
160 |
collection_name="nutritional_hypotheticals",
|
161 |
+
persist_directory="./nutritional_db",
|
162 |
+
#persist_directory="./nutritional_db2",
|
163 |
embedding_function=embedding_model
|
164 |
)
|
165 |
+
print("vector_store initialized...")
|
166 |
|
167 |
# Create a retriever from the vector store
|
168 |
retriever = vector_store.as_retriever(
|