Update app.py

app.py

@@ -38,7 +38,7 @@ prompt = ZeroShotAgent.create_prompt(
     suffix=suffix,
     input_variables=["input", "chat_history", "agent_scratchpad"],
 )
-memory = ConversationBufferMemory(memory_key="chat_history")
+memory = ConversationBufferMemory(memory_key="chat_history",return_messages=True)
 
 llm_chain = LLMChain(llm=llm, prompt=prompt)
 agent = ZeroShotAgent(llm_chain=llm_chain, tools=tools, verbose=True)
@@ -54,7 +54,7 @@ agent_chain = AgentExecutor.from_agent_and_tools(
 # )
 def chat(prompt):
     res = agent_chain.run(input=prompt)
-    return res,agent_chain.memory.chat_memory.messages
+    return res,memory.load_memory_variables({})
 
 iface = gr.Interface(
     fn=chat,