Change the Generate/Stop buttons

oobabooga · Sep 13, 2023 · 1b47b5c · 1b47b5c
1 parent 01335d4
commit 1b47b5c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/modules/ui_chat.py b/modules/ui_chat.py
@@ -34,8 +34,8 @@ def create_ui():
 
             with gr.Column(scale=1):
                 with gr.Row():
-                    shared.gradio['Stop'] = gr.Button('■', elem_id='stop', visible=False, variant='stop')
-                    shared.gradio['Generate'] = gr.Button('►', elem_id='Generate', variant='primary')
+                    shared.gradio['Stop'] = gr.Button('■', elem_id='stop', visible=False)
+                    shared.gradio['Generate'] = gr.Button('▶', elem_id='Generate', variant='primary')
 
         # Hover menu buttons
         shared.gradio['Regenerate'] = gr.Button('Regenerate')