Update app.py

app.py

@@ -26,8 +26,8 @@ POSTGRE_URL = os.environ['POSTGRE_URL']
 def get_llm_chain():
     return custom_chain_with_history(
         llm=CustomLLM(repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1", model_type='text-generation', api_token=API_TOKEN, stop=["\n<|","<|"], temperature=0.001), 
-        memory=st.session_state.memory.chat_memory,
-        # memory=st.session_state.memory
+        # memory=st.session_state.memory.chat_memory,
+        memory=st.session_state.memory
     )