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