Update UI and add 4:5

app.py

@@ -157,6 +157,8 @@ def get_image_size(aspect_ratio):
         return 1024, 688
     elif aspect_ratio == "2:3":
         return 688, 1024
+    elif aspect_ratio == "4:5":
+        return 1024, 1280
     else:
         # Default to 1:1 if something goes wrong
         return 1024, 1024
@@ -170,7 +172,7 @@ def infer(
     aspect_ratio="1:1",
     guidance_scale=1.0,
     num_inference_steps=8,
-    prompt_enhance=True,
+    prompt_enhance=False,
     progress=gr.Progress(track_tqdm=True),
 ):
     """
@@ -258,6 +260,8 @@ with gr.Blocks(css=css) as demo:
                         placeholder="Enter your prompt",
                         container=False,
                     )
+
+                with gr.Row():
                     run_button = gr.Button("Run", scale=0, variant="primary")
 
                 with gr.Accordion("Advanced Settings", open=False):
@@ -274,10 +278,10 @@ with gr.Blocks(css=css) as demo:
                     with gr.Row():
                         aspect_ratio = gr.Radio(
                             label="Aspect ratio (width:height)",
-                            choices=["1:1", "16:9", "9:16", "4:3", "3:4", "3:2", "2:3"],
+                            choices=["4:5", "1:1", "16:9", "9:16", "4:3", "3:4", "3:2", "2:3"],
                             value="16:9",
                         )
-                        prompt_enhance = gr.Checkbox(label="Prompt Enhance", value=True)
+                        prompt_enhance = gr.Checkbox(label="Prompt Enhance", value=False)
 
                     with gr.Row():
                         guidance_scale = gr.Slider(