Hugging Face's logo

Spaces:
DeepLabCut
/
MegaDetector_DeepLabCut
Build error

App Files Community
34
vchiang001 commited on
Commit
5c005de
•
1 Parent(s): 6940bf0

updated megadetector model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -247,7 +247,7 @@ def predict_pipeline(img_input,
247
  # Get MegaDetector model
248
  # TODO: Allow user selectable model?
249
  # models = ["model_weights/md_v5a.0.0.pt","model_weights/md_v5b.0.0.pt"]
250
- MD_model = torch.hub.load('ultralytics/yolov5', 'custom', "model_weights/md_v5a.0.0.pt")
251
 
252
 
253
  ####################################################
 
247
  # Get MegaDetector model
248
  # TODO: Allow user selectable model?
249
  # models = ["model_weights/md_v5a.0.0.pt","model_weights/md_v5b.0.0.pt"]
250
+ MD_model = torch.hub.load('ultralytics/yolov5', 'custom', "megadet_model/md_v5b.0.0.pt")
251
 
252
 
253
  ####################################################