Update app.py

app.py

@@ -24,7 +24,7 @@ import re
 
 
 
-# default_persist_directory = './chroma_HF/'
+default_persist_directory = './chroma_HF/'
 list_llm = ["mistralai/Mistral-7B-Instruct-v0.2"]#, "mistralai/Mixtral-8x7B-Instruct-v0.1", "mistralai/Mistral-7B-Instruct-v0.1", \
 #     "google/gemma-7b-it","google/gemma-2b-it", \
 #     "HuggingFaceH4/zephyr-7b-beta", "HuggingFaceH4/zephyr-7b-gemma-v0.1", \
@@ -60,7 +60,7 @@ def create_db(splits, collection_name):
         embedding=embedding,
         client=new_client,
         collection_name=collection_name,
-        # persist_directory=default_persist_directory
+        persist_directory=default_persist_directory
     )
     return vectordb
 
@@ -69,7 +69,7 @@ def create_db(splits, collection_name):
 def load_db():
     embedding = HuggingFaceEmbeddings()
     vectordb = Chroma(
-        # persist_directory=default_persist_directory, 
+        persist_directory=default_persist_directory, 
         embedding_function=embedding)
     return vectordb