Update app.py

app.py

@@ -6,7 +6,7 @@ torch.hub.download_url_to_file('https://cdn.pixabay.com/photo/2016/06/15/01/11/s
 torch.hub.download_url_to_file('https://cdn.pixabay.com/photo/2016/11/21/14/31/vw-bus-1845719_1280.jpg', 'bus.jpg')
 # Model
 model = torch.hub.load('ultralytics/yolov5', 'custom', 'customModel/model.pt')
-def yolo(im, size=1920):
+def yolo(im, size=640):
     g = (size / max(im.size))  # gain
     im = im.resize((int(x * g) for x in im.size), Image.ANTIALIAS)  # resize
     results = model(im)  # inference