Spaces:

neuratech-ai
/

person_segmentation_v3

Running

nlightcho commited on Jun 21, 2024

Commit

0136d74

verified ·

1 Parent(s): cc9455d

fix image resize

Files changed (1) hide show

app.py CHANGED Viewed

@@ -45,7 +45,7 @@ def query_image(img):
         return None
     img = Image.fromarray(img)
-    scale = 1024 / min(img.size)
     img = img.resize(
         (int(img.size[0] * scale), int(img.size[1] * scale)), Image.LANCZOS
     )

         return None
     img = Image.fromarray(img)
+    scale = 1024 / max(img.size)
     img = img.resize(
         (int(img.size[0] * scale), int(img.size[1] * scale)), Image.LANCZOS
     )