From 525a76b8365832220d58a3c38831ac1bc21d103b Mon Sep 17 00:00:00 2001 From: Tobias Cabanski Date: Tue, 28 Jan 2025 08:55:03 +0100 Subject: [PATCH] add kwargs to gradio launch (#367) --- src/smolagents/gradio_ui.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/smolagents/gradio_ui.py b/src/smolagents/gradio_ui.py index 52f952b..dc27c31 100644 --- 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"]