Update app.py

app.py

@@ -81,7 +81,7 @@ def response(message, history, image):
     for new_token in streamer:
         partial_response += new_token
         history[-1][1] = partial_response
-        yield history, gr.Button(visible=False), gr.Button(visible=True, interactive=True)
+        yield history
 
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
@@ -94,14 +94,14 @@ with gr.Blocks(css=css) as demo:
     response_handler = (
         response,
         [message, chat, image],
-        [submit]
+        [chat]
     )
 
-    postresponse_handler = (
-        lambda: (gr.Button(visible=False), gr.Button(visible=True)),
-        None,
-        [submit]
-    )
+    # postresponse_handler = (
+    #     lambda: (gr.Button(visible=False), gr.Button(visible=True)),
+    #     None,
+    #     [submit]
+    # )
 
     event = submit.click(*response_handler)
     # event.then(*postresponse_handler)