Update app.py

app.py

@@ -65,7 +65,7 @@ def my_inference_function(name):
 gradio_interface = gr.Interface(
   fn = chat,
   inputs = ["text", state, agent_state],
-  outputs = "text"
+  outputs = ["text", state]
 )
 gradio_interface.launch()