BhumikaMak commited on
Commit
32ce7f5
·
1 Parent(s): 136ee37

Fix: yolo8 loading issue.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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/yolov5:v7.0', 'yolov5', 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:
 
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: