Update app.py

app.py

@@ -20,7 +20,7 @@ import torch
 import tqdm 
 import accelerate
 
-default_persist_directory = '/ChromaDB'
+default_persist_directory = './ChromaDB'
 
 llm_name0 = "mistralai/Mixtral-8x7B-Instruct-v0.1"
 
@@ -55,9 +55,7 @@ def create_db(splits, collection_name):
 # Load vector database
 def load_db():
     embedding = HuggingFaceEmbeddings(model_name="sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2")
-    vectordb = Chroma(
-        # persist_directory=default_persist_directory, 
-        embedding_function=embedding)
+    vectordb = Chroma(persist_directory=default_persist_directory, embedding_function=embedding)
     return vectordb
 
 
@@ -110,7 +108,8 @@ def initialize_database(list_file_obj, chunk_size, chunk_overlap, progress=gr.Pr
     # Create or load Vector database
     progress(0.5, desc="Generating vector database...")
     # global vector_db
-    vector_db = create_db(doc_splits, collection_name)
+    #vector_db = create_db(doc_splits, collection_name)
+    vector_db = load_db()
     progress(0.9, desc="Done!")
     return vector_db, collection_name, "Complete!"