Spaces:

marsupial
/

marsupial

Runtime error

hlydecker commited on Sep 2, 2022

Commit

1cf1d3f

•

1 Parent(s): 8808f6b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ from PIL import Image
 # Load Marsupial model
 # TODO: Allow user selectable model?
-model = torch.hub.load('ultralytics/yolov5', 'custom', "model_weights/marsupial_72s_lures.pt", trust_repo=True)
 def yolo(im, size=640):
     g = (size / max(im.size))  # gain

 # Load Marsupial model
 # TODO: Allow user selectable model?
+# model = torch.hub.load('ultralytics/yolov5', 'custom', "model_weights/marsupial_72s_lures.pt", trust_repo=True)
 def yolo(im, size=640):
     g = (size / max(im.size))  # gain