Spaces:

tej11
/

object-detection

Runtime error

tej11 commited on Oct 31, 2022

Commit

b0c4e73

1 Parent(s): 8ed1223

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ import json
 # Model
 # model = torch.hub.load('ultralytics/yolov5', 'yolov5s')  # force_reload=True to update
-model = torch.hub.load('https://github.com/ultralytics/yolov5.git','custom',path='best.pt', source="local")
 def yolo(im, size=1024):
     g = (size / max(im.size))  # gain

 # Model
 # model = torch.hub.load('ultralytics/yolov5', 'yolov5s')  # force_reload=True to update
+model = torch.hub.load("best.pt")
 def yolo(im, size=1024):
     g = (size / max(im.size))  # gain