Update app/main.py

app/main.py

@@ -28,7 +28,7 @@ st.sidebar.html("""
 
 # Initialize chromadb variables
 EMBEDDING_MODEL = "all-MiniLM-L6-v2"
-model_path = "./chromadb_model"
+model_path = get_model_path()
 
 # Set-up Logger
 st.session_state.use_logger = False
@@ -76,7 +76,7 @@ else:
         process_pdf()  # Extract text from PDF
 
         with st.spinner("Extracting information from the text..."):
-            client, embedding_func = initialize_chromadb(EMBEDDING_MODEL)  
+            client, embedding_func = initialize_chromadb(EMBEDDING_MODEL, model_path)  
             whole_text_collection = initialize_collection(client, embedding_func, "whole_text_chunks")  
             update_collection(
                 whole_text_collection,