Update app.py

app.py

@@ -162,6 +162,10 @@ def preload_presets(target_ratio, ui_width, ui_height):
         changed_width = 1280
         changed_height = 720
         return changed_width, changed_height, gr.update(open=False)
+    elif target_ratio == "1:1":
+        changed_width = 1024
+        changed_height = 1024
+        return changed_width, changed_height, gr.update(open=False)
     elif target_ratio == "Custom":
         return ui_width, ui_height, gr.update(open=True)
 
@@ -213,7 +217,7 @@ with gr.Blocks(css=css) as demo:
                 with gr.Row():
                     target_ratio = gr.Radio(
                         label="Expected Ratio",
-                        choices=["9:16", "16:9", "Custom"],
+                        choices=["9:16", "16:9", "1:1", "Custom"],
                         value="9:16",
                         scale=2
                     )