Update app.py

app.py

@@ -1626,7 +1626,7 @@ if __name__ == "__main__":
     # GRADIO MODE
 
     @spaces.GPU()
-    def infer(prompt, resolution, num_inference_steps, guidance_scale, seed, progress=gr.Progress(track_tqdm=True)):
+    def infer(prompt, resolution, num_inference_steps, guidance_scale, seed, use_progressive_upscaling, progress=gr.Progress(track_tqdm=True)):
         set_seed(seed)
         width,height = list(map(int, resolution.split(',')))
         cross_attention_kwargs = {"edit_type": "visualize",
@@ -1660,7 +1660,7 @@ if __name__ == "__main__":
                       upscale_mode=args.upscale_mode, 
                       use_multidiffusion=args.use_multidiffusion, 
                       use_skip_residual=args.use_skip_residual, 
-                      use_progressive_upscaling=args.use_progressive_upscaling,
+                      use_progressive_upscaling=use_progressive_upscaling,
                       use_dilated_sampling=args.use_dilated_sampling,
                       shuffle=args.shuffle, 
                       result_path=result_path, 
@@ -1725,6 +1725,7 @@ if __name__ == "__main__":
                             num_inference_steps = gr.Slider(label="Inference Steps", minimum=2, maximum=50, step=1, value=50)
                             guidance_scale = gr.Slider(label="Guidance Scale", minimum=1, maximum=510, step=0.1, value=7.5)
                             seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=42)
+                    use_progressive_upscaling = gr.Checkbox(label="use_progressive_upscaling", value=False)
             
             output_images = gr.Image(label="Output Image", format="png")
             gr.Examples(
@@ -1737,7 +1738,7 @@ if __name__ == "__main__":
             )
         submit_btn.click(
             fn = infer,
-            inputs = [prompt, resolution, num_inference_steps, guidance_scale, seed],
+            inputs = [prompt, resolution, num_inference_steps, guidance_scale, seed, use_progressive_upscaling],
             outputs = [output_images],
             show_api=False
         )