Curse you, gradio!!! fixes broken refresh button #12309

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