Update app.py

app.py

@@ -5,9 +5,9 @@ from PIL import Image
 
 model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt', force_reload=True)
 
-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
+def yolo(im):
+    #g = (size / max(im.size))  # gain
+   # im = im.resize((int(x * g) for x in im.size), Image.ANTIALIAS)  # resize
 
     results = model(im)  # inference
     results.render()  # updates results.imgs with boxes and labels