Update app.py

app.py

@@ -19,16 +19,17 @@ colors = ["red",
           "slategray",
          ]
 
+# Resized image width
+WIDTH = 640
+
 def detect(image):
-    # Change width to 640 pixels
-    #h, w, c = image.shape
-    #scale = 640 / w
+    width, height = image.size
+    ratio = float(WIDTH) / float(w)
+    new_h = h * ratio
+
+    image = image.resize((WIDTH, new_h), Image.Resampling.LANCZOS)
     
-    inputs = image_processor(images=image, 
-                             do_resize=True,
-                             size={"shortest_edge": 480,
-                                   "longest_edge": 640},
-                             return_tensors="pt")
+    inputs = image_processor(images=image, return_tensors="pt")
     outputs = model(**inputs)
 
     # convert outputs to COCO API