add resizing

app.py

@@ -4,9 +4,27 @@ from transparent_background import Remover
 import numpy as np
 
 
+def resize(img: Image, target_size_px: int):
+    aspect_ratio = img.width / img.height
+
+    if img.width > img.height:
+        new_width = target_size_px
+        new_height = int(target_size_px / aspect_ratio)
+    else:
+        new_height = target_size_px
+        new_width = int(target_size_px * aspect_ratio)
+
+    img = img.resize((new_width, new_height), Image.ANTIALIAS)
+
+    return img
+
+
 def infer(img: Image):
+
+    img_resized = resize(img, 512)
+
     remover = Remover(mode="fast")
-    masked_image = remover.process(img, type="map")
+    masked_image = remover.process(img_resized, type="map")
 
     gray_image = masked_image.convert("L")
     binary_image = gray_image.point(lambda x: 0 if x < 128 else 255, "1")
@@ -22,10 +40,10 @@ def infer(img: Image):
     min_y = np.min(y_coords)
     max_y = np.max(y_coords)
 
-    draw = ImageDraw.Draw(img)
+    draw = ImageDraw.Draw(img_resized)
     draw.rectangle([(min_x, min_y), (max_x, max_y)], outline="red", width=2)
 
-    return img, masked_image
+    return img_resized, masked_image
 
 
 gr.Interface(

requirements.txt

@@ -1,2 +1,2 @@
-transparent-background = "*"
-numpy = "<2"
+transparent-background
+numpy<2