edjdhug3 commited on
Commit
588e2b3
·
1 Parent(s): 7eaf99d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -516,14 +516,14 @@ class Chatbot:
516
  db_instructEmbedd = FAISS.from_documents(self.docs, instructor_embeddings)
517
  self.retriever = db_instructEmbedd.as_retriever(search_kwargs={"k": 3})
518
 
519
- with open("db_instructEmbedd.pkl", "wb") as f:
520
- pickle.dump(db_instructEmbedd, f)
521
 
522
- def load_embeddings(self):
523
- with open("db_instructEmbedd.pkl", "rb") as f:
524
- self.retriever = pickle.load(f)
525
 
526
- self.retriever = self.retriever.as_retriever(search_kwargs={"k": 3})
527
 
528
  def create_qa_model(self):
529
  llm = HuggingFaceHub(repo_id="google/flan-t5-xxl", model_kwargs={"temperature": 0.1})
 
516
  db_instructEmbedd = FAISS.from_documents(self.docs, instructor_embeddings)
517
  self.retriever = db_instructEmbedd.as_retriever(search_kwargs={"k": 3})
518
 
519
+ # with open("db_instructEmbedd.pkl", "wb") as f:
520
+ # pickle.dump(db_instructEmbedd, f)
521
 
522
+ # def load_embeddings(self):
523
+ # with open("db_instructEmbedd.pkl", "rb") as f:
524
+ # self.retriever = pickle.load(f)
525
 
526
+ # self.retriever = self.retriever.as_retriever(search_kwargs={"k": 3})
527
 
528
  def create_qa_model(self):
529
  llm = HuggingFaceHub(repo_id="google/flan-t5-xxl", model_kwargs={"temperature": 0.1})