diff --git a/iz_helpers/ui.py b/iz_helpers/ui.py index 7cb0430..20d3922 100644 --- a/iz_helpers/ui.py +++ b/iz_helpers/ui.py @@ -249,22 +249,24 @@ 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) - 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 + 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, "", ""]),