diff options
| -rw-r--r-- | modules/ui_common.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/ui_common.py b/modules/ui_common.py index eefe0c0e..1dda1627 100644 --- a/modules/ui_common.py +++ b/modules/ui_common.py @@ -239,13 +239,13 @@ def create_refresh_button(refresh_component, refresh_method, refreshed_args, ele for comp in refresh_components:
setattr(comp, k, v)
- return [gr.update(**(args or {})) for _ in refresh_components]
+ return (gr.update(**(args or {})) for _ in refresh_components) if len(refresh_components) > 1 else gr.update(**(args or {}))
refresh_button = ToolButton(value=refresh_symbol, elem_id=elem_id, tooltip=f"{label}: refresh" if label else "Refresh")
refresh_button.click(
fn=refresh,
inputs=[],
- outputs=[*refresh_components]
+ outputs=refresh_components
)
return refresh_button
|