Update app.py

app.py

@@ -25,20 +25,25 @@ MAX_SEED = np.iinfo(np.int32).max
 
 examples = json.loads(open("examples.json").read())
 
-# (1664, 928), (1472, 1140), (1328, 1328)
+aspect_ratios = {
+    "Big 1:1": (1080, 1080),
+    "Big 16:9": (1920, 1080),
+    "Big 9:16": (1080, 1920),
+    "Big 4:3": (1440, 1080),
+    "Big 3:4": (1080, 1440),
+
+    "Small 1:1": (720, 720),
+    "Small 16:9": (1280, 720),
+    "Small 9:16": (720, 1280),
+    "Small 4:3": (960, 720),
+    "Small 3:4": (720, 960),
+}
+
 def get_image_size(aspect_ratio):
-    if aspect_ratio == "1:1":
-        return 1920, 1920
-    elif aspect_ratio == "16:9":
-        return 1920, 1080
-    elif aspect_ratio == "9:16":
-        return 1080, 1920
-    elif aspect_ratio == "4:3":
-        return 1920, 1440
-    elif aspect_ratio == "3:4":
-        return 1440, 1920
-    else:
-        return 640, 640
+
+
+
+    return aspect_ratios[aspect_ratio]
 
 def polish_prompt_en(original_prompt):
     
@@ -141,11 +146,13 @@ with gr.Blocks(css=css) as demo:
 
             with gr.Row():
                 aspect_ratio = gr.Radio(
-                    label="Image size (ratio, max dim 1920)",
-                    choices=["1:1", "16:9", "9:16", "4:3", "3:4"],
-                    value="16:9",
+                    label="Image size",
+                    choices=list(aspect_ratios.keys()),
+                    value="Small 4:3",
                 )
 
+
+
             with gr.Row():
                 guidance_scale = gr.Slider(
                     label="Guidance scale",