assistanttttttt commited on
Commit
2c31db4
·
1 Parent(s): 560df86

Fix run handler to duplicate output when gallery tab present

Browse files
Files changed (1) hide show
  1. ui/events/run_handlers.py +3 -1
ui/events/run_handlers.py CHANGED
@@ -101,7 +101,9 @@ def create_run_event(prefix: str, task_type: str, ui_components: dict):
101
  if gallery_tab:
102
  outputs.append(gallery_tab)
103
  run_btn.click(
104
- fn=lambda *args, progress=gr.Progress(track_tqdm=True): generate_image_wrapper(create_ui_inputs_dict(*args), progress),
 
 
105
  inputs=input_list_flat,
106
  outputs=outputs
107
  )
 
101
  if gallery_tab:
102
  outputs.append(gallery_tab)
103
  run_btn.click(
104
+ fn=lambda *args, progress=gr.Progress(track_tqdm=True): (
105
+ lambda res: [res, res] if ui_components.get('gallery') else res
106
+ )(generate_image_wrapper(create_ui_inputs_dict(*args), progress)),
107
  inputs=input_list_flat,
108
  outputs=outputs
109
  )