Update app.py
Browse files
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)
|