Update result gallery to use filepath type for correct handling

ui/events/run_handlers.py

@@ -104,9 +104,7 @@ def create_run_event(prefix: str, task_type: str, ui_components: dict):
         # No need to add it to the run button outputs.
 
         run_btn.click(
-            fn=lambda *args, progress=gr.Progress(track_tqdm=True): (
-                lambda res: [res, res] if ui_components.get('gallery') else res
-            )(generate_image_wrapper(create_ui_inputs_dict(*args), progress)),
+            fn=lambda *args, progress=gr.Progress(track_tqdm=True): generate_image_wrapper(create_ui_inputs_dict(*args), progress),
             inputs=input_list_flat,
             outputs=outputs
         )

ui/shared/txt2img_ui.py

@@ -86,7 +86,7 @@ def create_ui():
                     label="Result",
                     show_label=False,
                     columns=2,
-                    object_fit="contain",
+                    type="filepath",
                     height=627
                 )