add resolution settings

app.py

@@ -1625,9 +1625,9 @@ if __name__ == "__main__":
     # GRADIO MODE
 
     @spaces.GPU()
-    def infer(prompt, num_inference_steps, guidance_scale, progress=gr.Progress(track_tqdm=True)):
+    def infer(prompt, resolution, num_inference_steps, guidance_scale, progress=gr.Progress(track_tqdm=True)):
         set_seed(args.seed)
-        width,height = list(map(int, args.resolution.split(',')))
+        width,height = list(map(int, resolution.split(',')))
         cross_attention_kwargs = {"edit_type": "visualize",
                                   "n_self_replace": 0.4,
                                   "n_cross_replace": {"default_": 1.0, "confetti": 0.8},
@@ -1690,6 +1690,13 @@ if __name__ == "__main__":
             """)
             prompt = gr.Textbox(label="Prompt")
             with gr.Accordion("Advanced settings", open=False):
+                resolution = gr.Radio(
+                    label = "Resolution",
+                    choices = [
+                        "1024,1024", "2048,2048", "2048,1024", "1536,3072", "3072,3072", "4096,4096", "4096,2048"
+                    ],
+                    value = "1024,1024"
+                )
                 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)
             submit_btn = gr.Button("Submit")
@@ -1704,7 +1711,7 @@ if __name__ == "__main__":
             )
         submit_btn.click(
             fn = infer,
-            inputs = [prompt, num_inference_steps, guidance_scale],
+            inputs = [prompt, resolution, num_inference_steps, guidance_scale],
             outputs = [output_images],
             show_api=False
         )