Spaces:

JosephTK
/

object-detection-count

Sleeping

JosephTK commited on May 26, 2023

Commit

f158078

•

1 Parent(s): 64377f8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -27,7 +27,7 @@ def detect(image):
     ratio = float(WIDTH) / float(width)
     new_h = height * ratio
-    image = image.resize((WIDTH, new_h), Image.Resampling.LANCZOS)
     inputs = image_processor(images=image, return_tensors="pt")
     outputs = model(**inputs)

     ratio = float(WIDTH) / float(width)
     new_h = height * ratio
+    image = image.resize((int(WIDTH), int(new_h)), Image.Resampling.LANCZOS)
     inputs = image_processor(images=image, return_tensors="pt")
     outputs = model(**inputs)