Update app.py

app.py

@@ -5,7 +5,7 @@ import gradio as gr
 import pickle
 from query_data import get_chain
 from threading import Lock
-from ingest_data import ingest_data
+from ingest_data import get_vectorstore
 
 
 
@@ -21,9 +21,9 @@ class ChatWrapper:
         """Execute the chat functionality."""
         self.lock.acquire()
         try:
-            vs = ingest_data("vectorstore.pkl")
-            # with open(vs, "rb") as f:
-            #     vectorstore = pickle.load(f)
+            vs = get_vectorstore("vectorstore.pkl")
+            with open(vs, "rb") as f:
+                vectorstore = pickle.load(f)
                 
             history = history or []
             chain = get_chain(vectorstore)