Update app.py

app.py

@@ -223,7 +223,7 @@ def create_chain(llm, prompt, CONDENSE_QUESTION_PROMPT, db):
     Returns:
         ConversationalRetrievalChain: The initialized conversational chain.
     """
-    memory = ConversationTokenBufferMemory(llm=llm, memory_key="chat_history", return_messages=True, input_key='question')
+    memory = ConversationTokenBufferMemory(llm=llm, memory_key="chat_history", return_messages=True, input_key='question', output_key='answer')
     chain = ConversationalRetrievalChain.from_llm(
         llm=llm,
         chain_type="stuff",