Update app.py

app.py

@@ -58,6 +58,9 @@ def update_selection(evt: gr.SelectData, width, height):
         elif selected_lora["aspect"] == "landscape":
             width = 1024
             height = 768
+        else:
+            width = 1024
+            height = 1024
     return (
         gr.update(placeholder=new_placeholder),
         updated_text,