Update app.py

app.py

@@ -128,7 +128,8 @@ def main():
                     st.write(end - start)
     
                     st.write("Bot Response:")
-                    st.write(agent.invoke(query, config={"callbacks":[ConsoleCallbackHandler()]}))
+                    # st.write(agent.invoke(query, config={"callbacks":[ConsoleCallbackHandler()]}))
+                    st.write(agent.run({"input": query}
                     # st.write(response)