Update app.py

app.py

@@ -516,14 +516,14 @@ class Chatbot:
         db_instructEmbedd = FAISS.from_documents(self.docs, instructor_embeddings)
         self.retriever = db_instructEmbedd.as_retriever(search_kwargs={"k": 3})
 
-    #     with open("db_instructEmbedd.pkl", "wb") as f:
-    #         pickle.dump(db_instructEmbedd, f)
+        with open("db_instructEmbedd.pkl", "wb") as f:
+            pickle.dump(db_instructEmbedd, f)
 
-    # def load_embeddings(self):
-    #     with open("db_instructEmbedd.pkl", "rb") as f:
-    #         self.retriever = pickle.load(f)
+    def load_embeddings(self):
+        with open("db_instructEmbedd.pkl", "rb") as f:
+            self.retriever = pickle.load(f)
 
-    #     self.retriever = self.retriever.as_retriever(search_kwargs={"k": 3})
+        self.retriever = self.retriever.as_retriever(search_kwargs={"k": 3})
 
     def create_qa_model(self):
         llm = HuggingFaceHub(repo_id="google/flan-t5-xxl", model_kwargs={"temperature": 0.1})