Update app.py

app.py

@@ -16,7 +16,7 @@ models = ["model_weights/md_v5a.0.0.pt","model_weights/md_v5b.0.0.pt"]
 
 # model = torch.hub.load('ultralytics/yolov5', 'custom', "model_weights/md_v5a.0.0.pt")
 
-def yolo(im, size=640, model_name):
+def yolo(img_input, size=640, model_name):
 
     model = model = torch.hub.load('ultralytics/yolov5', 'custom', model_name)
     g = (size / max(im.size))  # gain