vishwask commited on
Commit
ccb265b
·
verified ·
1 Parent(s): 66bbfcf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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(llm_model,retriever=retriever,chain_type="stuff",
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')