Update app.py

app.py

@@ -25,11 +25,11 @@ def main(img):
     # Checking Image Aspect Ratio
     #load_img = Image.open(img)
     #image size
-    #ratio_width = load_img.size[0]
+    ratio_width, ratio_height = img.size
     #ratio_height = load_img.size[1]
-    #print(f"Aspect ratio: {ratio_width}x{ratio_height} px")
-    #if ratio_width >= 1026 or ratio_height >= 1026:
-    #    raise gr.Error("Image aspect ratio must not exceed width: 1024 px and height: 1024 px.")
+    print(f"Aspect ratio: {ratio_width}x{ratio_height} px")
+    if ratio_width >= 1026 or ratio_height >= 1026:
+        raise gr.Error("Image aspect ratio must not exceed width: 1024 px and height: 1024 px.")
 
     # setup folder and path
     #basewidth = 256