vishwask commited on
Commit
323d793
·
verified ·
1 Parent(s): 2b9fe40

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -129,12 +129,13 @@ def initialize_llmchain(temperature, max_tokens, top_k, vector_db, progress=gr.P
129
 
130
  progress(0.75, desc="Defining buffer memory...")
131
  #memory = ConversationBufferMemory(memory_key="chat_history",output_key='answer',return_messages=True)
132
- memory = ConversationTokenBufferMemory(llm = llm, max_token_limit=100)
133
  # retriever=vector_db.as_retriever(search_type="similarity", search_kwargs={'k': 3})
134
  retriever=vector_db.as_retriever()
135
  progress(0.8, desc="Defining retrieval chain...")
136
  qa_chain = ConversationalRetrievalChain.from_llm(llm,retriever=retriever,chain_type="stuff",
137
- memory=memory,return_source_documents=True,verbose=False)
 
138
  progress(0.9, desc="Done!")
139
  return qa_chain
140
 
 
129
 
130
  progress(0.75, desc="Defining buffer memory...")
131
  #memory = ConversationBufferMemory(memory_key="chat_history",output_key='answer',return_messages=True)
132
+ #memory = ConversationTokenBufferMemory(llm = llm, max_token_limit=100)
133
  # retriever=vector_db.as_retriever(search_type="similarity", search_kwargs={'k': 3})
134
  retriever=vector_db.as_retriever()
135
  progress(0.8, desc="Defining retrieval chain...")
136
  qa_chain = ConversationalRetrievalChain.from_llm(llm,retriever=retriever,chain_type="stuff",
137
+ #memory=memory,
138
+ return_source_documents=True,verbose=False)
139
  progress(0.9, desc="Done!")
140
  return qa_chain
141