Jatinydv commited on
Commit
7600365
1 Parent(s): 88f03bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -50,7 +50,7 @@ def load_llm():
50
  def qa_bot(query):
51
  embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2",
52
  model_kwargs={'device': 'cpu'})
53
- db = FAISS.load_local(DB_FAISS_PATH, embeddings)
54
  llm = load_llm()
55
  qa_prompt = set_custom_prompt()
56
  qa = retrieval_qa_chain(llm, qa_prompt, db)
 
50
  def qa_bot(query):
51
  embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2",
52
  model_kwargs={'device': 'cpu'})
53
+ db = FAISS.load_local(DB_FAISS_PATH, embeddings,allow_dangerous_deserialization=True)
54
  llm = load_llm()
55
  qa_prompt = set_custom_prompt()
56
  qa = retrieval_qa_chain(llm, qa_prompt, db)