Update app.py

app.py

@@ -76,7 +76,7 @@ if prompt := st.chat_input("What is up?"):
 
     # Display assistant response in chat message container
     with st.chat_message("assistant"):
-        response = qa_memory({"question": prompt})['result']
+        response = qa_memory({"question": prompt})
         st.write(response)
     # Add assistant response to chat history
     st.session_state.messages.append({"role": "assistant", "content": response})