i

app.py

@@ -366,7 +366,8 @@ with gr.Blocks(
                 undo_btn=undo_btn,
                 clear_btn=clear_btn,
                 additional_inputs=[seed, temperature, top_p, max_new_tokens],
-                examples=examples
+                examples=examples,
+                show_stop_button=False
             )
 
     render_button.click(

chat_interface.py

@@ -57,6 +57,7 @@ class ChatInterface(Blocks):
         pre_fn: Callable,
         chatbot: Chatbot,
         *,
+        show_stop_button=True,
         post_fn_kwargs: dict = None,
         pre_fn_kwargs: dict = None,
         multimodal: bool = False,
@@ -103,6 +104,8 @@ class ChatInterface(Blocks):
         self.pre_fn = pre_fn
         self.pre_fn_kwargs = pre_fn_kwargs
 
+        self.show_stop_button = show_stop_button
+
         self.interrupter = State(None)
 
         self.multimodal = multimodal
@@ -409,7 +412,7 @@ class ChatInterface(Blocks):
                         async_lambda(
                             lambda: (
                                 Button(visible=False),
-                                Button(visible=True),
+                                Button(visible=self.show_stop_button),
                             )
                         ),
                         None,
@@ -427,7 +430,7 @@ class ChatInterface(Blocks):
             else:
                 for event_trigger in event_triggers:
                     event_trigger(
-                        async_lambda(lambda: Button(visible=True)),
+                        async_lambda(lambda: Button(visible=self.show_stop_button)),
                         None,
                         [self.stop_btn],
                         show_api=False,