Update app.py

app.py

@@ -158,7 +158,7 @@ with gr.Blocks() as demo:
         with gr.Column(scale=3):
             # Custom chat function wrapper to update state
             def chat_with_state(message, history, state, temperature, top_p):
-                for partial_response, updated_state in predict(message, history, temperature, top_p):
+                for partial_response in predict(message, history, temperature, top_p):
                     # Update our state with each yield
                     state = history.copy()
                     yield partial_response, state