Spaces:
Running
Running
Commit
·
32ce7f5
1
Parent(s):
136ee37
Fix: yolo8 loading issue.
Browse files
app.py
CHANGED
@@ -45,7 +45,7 @@ def load_yolo_model(version="yolov5"):
|
|
45 |
if version == "yolov5":
|
46 |
model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True)
|
47 |
elif version == "yolov8":
|
48 |
-
model = torch.hub.load('ultralytics/
|
49 |
elif version == "yolov10":
|
50 |
model = torch.hub.load('ultralytics/yolov5', 'yolov5m', pretrained=True) # Placeholder for YOLOv10 (use an appropriate version if available)
|
51 |
else:
|
|
|
45 |
if version == "yolov5":
|
46 |
model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True)
|
47 |
elif version == "yolov8":
|
48 |
+
model = torch.hub.load('ultralytics/yolov8', 'yolov8s', pretrained=True) # YOLOv8 is part of the yolov5 repo starting from v7.0
|
49 |
elif version == "yolov10":
|
50 |
model = torch.hub.load('ultralytics/yolov5', 'yolov5m', pretrained=True) # Placeholder for YOLOv10 (use an appropriate version if available)
|
51 |
else:
|