Commit ·
2c31db4
1
Parent(s): 560df86
Fix run handler to duplicate output when gallery tab present
Browse files
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):
|
|
|
|
|
|
|
| 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 |
)
|