diff --git a/iz_helpers/ui.py b/iz_helpers/ui.py
index 0c6062c..7cb0430 100644
--- a/iz_helpers/ui.py
+++ b/iz_helpers/ui.py
@@ -249,16 +249,22 @@ Our best experience and trade-off is the R-ERSGAn4x upscaler.
 """
                         )
 
-            with gr.Column(scale=1, variant="compact"):
-                output_video = gr.Video(label="Output").style(width=512, height=512)
-                (
-                    out_image,
-                    generation_info,
-                    html_info,
-                    html_log,
-                ) = create_output_panel(
-                    "infinite-zoom", shared.opts.outdir_img2img_samples
-                )
+                with gr.Column(scale=1, variant="compact"):
+                    output_video = gr.Video(label="Output").style(width=512, height=512)
+                    output_panel = create_output_panel("infinite-zoom", shared.opts.outdir_img2img_samples)
+                    
+                if isinstance(output_panel, tuple):
+                    (
+                        out_image,
+                        generation_info,
+                        html_info,
+                        html_log,
+                    ) = output_panel
+                else:
+                    out_image = output_panel.gallery
+                    generation_info = output_panel.generation_info
+                    html_info = output_panel.infotext
+                    html_log = output_panel.html_log
 
         generate_btn.click(
             fn=wrap_gradio_gpu_call(create_zoom, extra_outputs=[None, "", ""]),