KeyArgo
/
smolagents
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add kwargs to gradio launch (#367)

This commit is contained in:
Tobias Cabanski 2025-01-28 08:55:03 +01:00 committed by GitHub
parent dca7081394
commit 525a76b836
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/smolagents/gradio_ui.py
View File

@ -173,7 +173,7 @@ class GradioUI:
"", "",
) )
def launch(self): def launch(self, **kwargs):
import gradio as gr import gradio as gr
with gr.Blocks() as demo: with gr.Blocks() as demo:
@ -204,7 +204,7 @@ class GradioUI:
[stored_messages, text_input], [stored_messages, text_input],
).then(self.interact_with_agent, [stored_messages, chatbot], [chatbot]) ).then(self.interact_with_agent, [stored_messages, chatbot], [chatbot])
demo.launch() demo.launch(**kwargs)
__all__ = ["stream_to_gradio", "GradioUI"] __all__ = ["stream_to_gradio", "GradioUI"]