Update app.py

app.py

@@ -131,8 +131,8 @@ 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 = ConversationBufferWindowMemory(memory_key = 'history', k=3)
+    #memory = ConversationBufferMemory(memory_key="chat_history",output_key='answer',return_messages=True)
+    memory = ConversationBufferWindowMemory(memory_key = 'chat_history', k=3,output_key='answer',return_messages=True)
     retriever=vector_db.as_retriever()
     progress(0.8, desc="Defining retrieval chain...")
     qa_chain = ConversationalRetrievalChain.from_llm(llm,retriever=retriever,chain_type="stuff",