Update app.py
Browse files
app.py
CHANGED
@@ -153,7 +153,7 @@ llm = HuggingFaceHub(repo_id = llm_model,
|
|
153 |
"load_in_8bit": True})
|
154 |
retriever=vector_db.as_retriever()
|
155 |
memory = ConversationBufferMemory(memory_key="chat_history", output_key='answer', return_messages=True)
|
156 |
-
qa_chain = ConversationalRetrievalChain.from_llm(
|
157 |
memory=memory,return_source_documents=True,verbose=False,)
|
158 |
|
159 |
print('qa chain and vector_db done')
|
|
|
153 |
"load_in_8bit": True})
|
154 |
retriever=vector_db.as_retriever()
|
155 |
memory = ConversationBufferMemory(memory_key="chat_history", output_key='answer', return_messages=True)
|
156 |
+
qa_chain = ConversationalRetrievalChain.from_llm(llm,retriever=retriever,chain_type="stuff",
|
157 |
memory=memory,return_source_documents=True,verbose=False,)
|
158 |
|
159 |
print('qa chain and vector_db done')
|