Spaces:
Runtime error
Runtime error
Update app.py
Browse filesmodificación del path
app.py
CHANGED
@@ -42,7 +42,7 @@ torch.hub.download_url_to_file('https://i.pinimg.com/originals/c2/ce/e0/c2cee056
|
|
42 |
# Model
|
43 |
#model = torch.hub.load('ultralytics/yolov5', 'yolov5s') # force_reload=True to update
|
44 |
|
45 |
-
model = torch.hub.load('ultralytics/yolov5', 'custom', path='
|
46 |
#model = torch.hub.load('path/to/yolov5', 'custom', path='/content/yolov56.pt', source='local') # local repo
|
47 |
|
48 |
def yolo(im, size=640):
|
|
|
42 |
# Model
|
43 |
#model = torch.hub.load('ultralytics/yolov5', 'yolov5s') # force_reload=True to update
|
44 |
|
45 |
+
model = torch.hub.load('ultralytics/yolov5', 'custom', path='./best.pt') # local model o google colab
|
46 |
#model = torch.hub.load('path/to/yolov5', 'custom', path='/content/yolov56.pt', source='local') # local repo
|
47 |
|
48 |
def yolo(im, size=640):
|