Spaces:

DeepLabCut
/

MegaDetector_DeepLabCut

Build error

vchiang001 commited on Aug 9, 2022

Commit

ec768a3

•

1 Parent(s): 8966749

change model directory

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,14 +19,14 @@ model = torch.hub.load('ultralytics/yolov5', 'custom', "https://huggingface.co/s
 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', "/home/vic/vic_data/dlclive4mega/models/md_v5b.0.0.pt", force_reload=True)
     results = model(im)  # inference
     results.render()  # updates results.imgs with boxes and labels
     return Image.fromarray(results.imgs[0])  """
 def yolo(im): #size=640):
-    model = torch.hub.load('ultralytics/yolov5', 'custom', "/home/vic/vic_data/dlclive4mega/models/md_v5b.0.0.pt", force_reload=True)
     results = model(im)  # inference
     results.render()  # updates results.imgs with boxes and labels
     return Image.fromarray(results.imgs[0])

 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', "https://huggingface.co/spaces/vchiang001/MegaDetector_DLC_pose/blob/main/md_v5b.0.0.pt", force_reload=True)
     results = model(im)  # inference
     results.render()  # updates results.imgs with boxes and labels
     return Image.fromarray(results.imgs[0])  """
 def yolo(im): #size=640):
+    model = torch.hub.load('ultralytics/yolov5', 'custom', "https://huggingface.co/spaces/vchiang001/MegaDetector_DLC_pose/blob/main/md_v5b.0.0.pt", force_reload=True)
     results = model(im)  # inference
     results.render()  # updates results.imgs with boxes and labels
     return Image.fromarray(results.imgs[0])