Update app.py

app.py

@@ -29,13 +29,16 @@ class APIClient:
         return self._post(f"{self.base_url}/v2/job", json=body)
     
 
-def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=5.0, num_inference_steps=28, progress=gr.Progress(track_tqdm=True)):
+def infer(prompt, seed=42, randomize_seed=False, resolution="1024x1024", guidance_scale=5.0, num_inference_steps=28, progress=gr.Progress(track_tqdm=True)):
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
+
+    width, height = resolution.split("x")
+        
     image = generative_api.job({
         "prompt": prompt,
-        "width": width,
-        "height": height,
+        "width": int(width),
+        "height": int(height),
         "seed": seed,
         "num_inference_steps": num_inference_steps,
         "guidance_scale": guidance_scale
@@ -114,22 +117,14 @@ with gr.Blocks(css=css) as demo:
             
             with gr.Row():
                 
-                width = gr.Slider(
-                    label="Width",
-                    minimum=256,
-                    maximum=MAX_IMAGE_SIZE,
-                    step=32,
-                    value=1024,
-                    interactive=False
-                )
-                                                    
-                height = gr.Slider(
-                    label="Height",
-                    minimum=256,
-                    maximum=MAX_IMAGE_SIZE,
-                    step=32,
-                    value=1024,
-                    interactive=False
+                resolution = gr.Dropdown(
+                    label="Resolution",
+                    value="1024x1024",
+                    choices=[
+                        "1024x1024",
+                        "1024x576",
+                        "576x1024"
+                    ]
                 )
             
             with gr.Row():
@@ -161,7 +156,7 @@ with gr.Blocks(css=css) as demo:
     gr.on(
         triggers=[run_button.click, prompt.submit],
         fn = infer,
-        inputs = [prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
+        inputs = [prompt, seed, randomize_seed, resolution, guidance_scale, num_inference_steps],
         outputs = [result, seed]
     )