Update app.py

app.py

@@ -128,14 +128,13 @@ def initialize_llmchain(temperature, max_tokens, top_k, vector_db, progress=gr.P
                                                           "load_in_8bit": True})
     
     progress(0.75, desc="Defining buffer memory...")
-    #memory = ConversationBufferMemory(memory_key="chat_history",output_key='answer',return_messages=True)
+    memory = ConversationBufferMemory(memory_key="chat_history",output_key='answer',return_messages=True)
     #memory = ConversationTokenBufferMemory(llm = llm, max_token_limit=100)
     # retriever=vector_db.as_retriever(search_type="similarity", search_kwargs={'k': 3})
     retriever=vector_db.as_retriever()
     progress(0.8, desc="Defining retrieval chain...")
     qa_chain = ConversationalRetrievalChain.from_llm(llm,retriever=retriever,chain_type="stuff", 
-                                                     #memory=memory,
-                                                     return_source_documents=True,verbose=False)
+                                                     memory=memory,return_source_documents=True,verbose=False)
     progress(0.9, desc="Done!")
     return qa_chain
 
@@ -221,9 +220,6 @@ def upload_file(file_obj):
     # initialize_database(file_path, progress)
     return list_file_path
 
-
-
-
 def demo():
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
         vector_db = gr.State()