Update app.py

app.py

@@ -59,6 +59,15 @@ def generate(image_editor, prompt, neg_prompt, versions):
                            image=image, 
                            mask_image=mask).images[0]
 
+
+    if (original_image_size[0] > original_image_size[1]):
+        original_image_size[0] = original_image_size[0] * (1024/original_image_size[0])
+        original_image_size[1] = original_image_size[1] * (1024/original_image_size[0])
+    else:
+        original_image_size[0] = original_image_size[0] * (1024/original_image_size[1])
+        original_image_size[1] = original_image_size[1] * (1024/original_image_size[1])
+
+    
     # Crop image to original aspect ratio
     final_image = final_image.crop((0, 0, original_image_size[0], original_image_size[1]))