Hugging Face's logo

Spaces:
hlydecker
/
MegaDetector_v5
Runtime error

App Files Community
hlydecker commited on
Commit
5924b60
·
1 Parent(s): 74489e9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ article = "<p style='text-align: center'>MegaDetector makes predictions using a
14
  # TODO: Allow user selectable model?
15
  models = ["model_weights/md_v5a.0.0.pt","model_weights/md_v5b.0.0.pt"]
16
 
17
- model = torch.hub.load('ultralytics/yolov5', 'custom', "model_weights/md_v5a.0.0.pt")
18
 
19
  def yolo(im, size=640, model_name):
20
 
 
14
  # TODO: Allow user selectable model?
15
  models = ["model_weights/md_v5a.0.0.pt","model_weights/md_v5b.0.0.pt"]
16
 
17
+ # model = torch.hub.load('ultralytics/yolov5', 'custom', "model_weights/md_v5a.0.0.pt")
18
 
19
  def yolo(im, size=640, model_name):
20