diff --git a/scripts/infinite-zoom.py b/scripts/infinite-zoom.py
index 945f674..8754d6e 100644
--- a/scripts/infinite-zoom.py
+++ b/scripts/infinite-zoom.py
@@ -640,7 +640,7 @@ Our best experience and trade-off is the R-ERSGAn4x upscaler.
                     "infinite-zoom", shared.opts.outdir_img2img_samples
                 )
         generate_btn.click(
-            fn=create_zoom,
+            fn=wrap_gradio_gpu_call(create_zoom, extra_outputs=[None, '', '']),
             inputs=[
                 main_prompts,
                 main_negative_prompt,