Update app.py

app.py

@@ -19,8 +19,15 @@ colors = ["red",
           "slategray",
          ]
 
-def detect(image): 
-    inputs = image_processor(images=image, return_tensors="pt")
+def detect(image):
+    # Change width to 640 pixels
+    #h, w, c = image.shape
+    #scale = 640 / w
+    
+    inputs = image_processor(images=image, 
+                             do_resize=True,
+                             size={"longest_edge": 640},
+                             return_tensors="pt")
     outputs = model(**inputs)
 
     # convert outputs to COCO API
@@ -64,7 +71,7 @@ def detect(image):
 demo = gr.Interface(
     fn=detect,
     examples=["examples/football.jpg", "examples/cats.jpg"],
-    inputs=[gr.inputs.Image(label="Input image", type="pil", shape=(400, 360))],
+    inputs=[gr.inputs.Image(label="Input image", type="pil"],
     outputs=[gr.Image(label="Output image"), gr.BarPlot(show_label=False, x="label", y="counts", x_title="Labels", y_title="Counts", vertical=False), gr.Textbox(show_label=False)],
     title="YOLO Object Detection",
 )