add kwargs to gradio launch (#367)

2025-01-28 08:55:03 +01:00 · 2025-01-28 08:55:03 +01:00 · 525a76b836
parent dca7081394
commit 525a76b836
1 changed files with 2 additions and 2 deletions
--- a/src/smolagents/gradio_ui.py
+++ b/src/smolagents/gradio_ui.py
@ -173,7 +173,7 @@ class GradioUI:
            "",
        )

-    def launch(self):
+    def launch(self, **kwargs):
        import gradio as gr

        with gr.Blocks() as demo:
@ -204,7 +204,7 @@ class GradioUI:
                [stored_messages, text_input],
            ).then(self.interact_with_agent, [stored_messages, chatbot], [chatbot])

-        demo.launch()
+        demo.launch(**kwargs)


 __all__ = ["stream_to_gradio", "GradioUI"]