Update gradio_demo/app.py

gradio_demo/app.py

@@ -114,7 +114,7 @@ def dynamic_guidance_slider(sampling_steps):
 def show_final_preview(preview_row):
     return preview_row[-1][0]
 
-@spaces.GPU #[uncomment to use ZeroGPU]
+@spaces.GPU(duration=70) #[uncomment to use ZeroGPU]
 @torch.no_grad()
 def instantir_restore(
     lq, prompt="", steps=30, cfg_scale=7.0, guidance_end=1.0,
@@ -134,7 +134,13 @@ def instantir_restore(
         preview_start = preview_start / steps
     elif preview_start > 1.0:
         preview_start = preview_start / steps
-    lq = [resize_img(lq.convert("RGB"), size=None)]
+
+    w, h = lq.size
+    if w == h :
+        lq = [resize_img(lq.convert("RGB"), size=(width, heigth))]
+    else:
+        lq = [resize_img(lq.convert("RGB"), size=None)]
+   
     generator = torch.Generator(device=device).manual_seed(seed)
     timesteps = [
         i * (1000//steps) + pipe.scheduler.config.steps_offset for i in range(0, steps)