Update app.py

app.py

@@ -20,13 +20,13 @@ def setup_model(args):
     
     return model
 
-def main(img):
+def main(img: Image):
 
     # Checking Image Aspect Ratio
-    with Image.open(img) as im:
-        #image size
-        ratio_width = im.size[0]
-        ratio_height = im.size[1]
+    load_img = Image.fromarray(img)
+    #image size
+    ratio_width = load_img.size[0]
+    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.")