Update app.py

app.py

@@ -17,7 +17,7 @@ def download_image(url):
 hf_token = os.environ.get("HF_TOKEN_API_DEMO") # we get it from a secret env variable, such that it's private
 auth_headers = {"api_token": hf_token}
 
-resolutions = ["1024 1024","1280 768","1344 768","768 1344","768 1280"] 
+aspect_ratios = ["1:1","2:3","3:2","3:4","4:3","4:5","5:4","9:16","16:9"] 
 
 # Ng
 default_negative_prompt= "Logo,Watermark,Text,Ugly,Morbid,Extra fingers,Poorly drawn hands,Mutation,Blurry,Extra limbs,Gross proportions,Missing arms,Mutated hands,Long neck,Duplicate,Mutilated,Mutilated hands,Poorly drawn face,Deformed,Bad anatomy,Cloned face,Malformed limbs,Missing legs,Too many fingers"
@@ -28,7 +28,7 @@ default_negative_prompt= "Logo,Watermark,Text,Ugly,Morbid,Extra fingers,Poorly d
 # pipe.to(device="cuda")
 
 # @spaces.GPU(enable_queue=True)
-def infer(prompt,negative_prompt,seed,resolution):
+def infer(prompt,negative_prompt,seed,aspect_ratio):
     print(f"""
     —/n
     {prompt}
@@ -46,8 +46,7 @@ def infer(prompt,negative_prompt,seed,resolution):
         except:
             generator=None
 
-    w,h = resolution.split()
-    w,h = int(w),int(h)
+
     # image = pipe(prompt,num_inference_steps=30, negative_prompt=negative_prompt,generator=generator,width=w,height=h).images[0]
     url = "http://engine.int.bria-api.com/v1/text-to-image/base/3.2"
     
@@ -62,7 +61,8 @@ def infer(prompt,negative_prompt,seed,resolution):
     "include_generation_prefix": False,
     "negative_prompt": negative_prompt,
     "num_inference_steps": 30,
-    "seed": seed
+    "seed": seed,
+    "aspect_ratio": aspect_ratio
     })
     response = requests.request("POST", url, headers=auth_headers, data=payload)
     print('1',response)
@@ -104,7 +104,7 @@ with gr.Blocks(css=css) as demo:
         with gr.Group():
             with gr.Column():
                 prompt_in = gr.Textbox(label="Prompt", value='''photo of mystical dragon eating sushi, text bubble says "Sushi Time".''')
-                resolution = gr.Dropdown(value=resolutions[0], show_label=True, label="Resolution", choices=resolutions)
+                aspect_ratio = gr.Dropdown(value=aspect_ratios[0], show_label=True, label="Aspect Ratio", choices=aspect_ratios)
                 seed = gr.Textbox(label="Seed", value=-1)
                 negative_prompt = gr.Textbox(label="Negative Prompt", value=default_negative_prompt)
                 submit_btn = gr.Button("Generate")
@@ -133,7 +133,7 @@ with gr.Blocks(css=css) as demo:
             prompt_in,
             negative_prompt,
             seed,
-            resolution
+            aspect_ratio
         ],
         outputs = [
             result