vishwask commited on
Commit
398aeb6
·
verified ·
1 Parent(s): b974464

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -134,9 +134,7 @@ def initialize_llmchain(temperature, max_tokens, top_k, vector_db, progress=gr.P
134
  memory = ConversationBufferWindowMemory(memory_key = 'chat_history', k=3,output_key='answer',return_messages=True)
135
  retriever=vector_db.as_retriever()
136
  progress(0.8, desc="Defining retrieval chain...")
137
- # qa_chain = ConversationalRetrievalChain.from_llm(llm,retriever=retriever,chain_type="stuff",
138
- # memory=memory,return_source_documents=True,verbose=False)
139
- qa_chain = ConversationalRetrievalChain.from_llm(llm,retriever=retriever,chain_type="refine",
140
  memory=memory,return_source_documents=True,verbose=False)
141
  progress(0.9, desc="Done!")
142
  return qa_chain
 
134
  memory = ConversationBufferWindowMemory(memory_key = 'chat_history', k=3,output_key='answer',return_messages=True)
135
  retriever=vector_db.as_retriever()
136
  progress(0.8, desc="Defining retrieval chain...")
137
+ qa_chain = ConversationalRetrievalChain.from_llm(llm,retriever=retriever,chain_type="stuff",
 
 
138
  memory=memory,return_source_documents=True,verbose=False)
139
  progress(0.9, desc="Done!")
140
  return qa_chain