Update app.py

app.py

@@ -8,13 +8,13 @@ from PIL import Image
 # Load MegaDetector v5a model
 # TODO: Allow user selectable model?
 # 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")
+model = torch.hub.load('ultralytics/yolov5:v6.2', 'custom', "model_weights/md_v5a.0.0.pt")
 
 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
     
-    model = torch.hub.load('ultralytics/yolov5', 'custom', "model_weights/md_v5a.0.0.pt")
+    model = torch.hub.load('ultralytics/yolov5:v6.2', 'custom', "model_weights/md_v5a.0.0.pt")
     
     results = model(im)  # inference
     results.render()  # updates results.imgs with boxes and labels