babaTEEpe commited on
Commit
37265d8
·
verified ·
1 Parent(s): ee86132

Update mission_models.py

Browse files
Files changed (1) hide show
  1. mission_models.py +1 -1
mission_models.py CHANGED
@@ -141,7 +141,7 @@ class ObjectDetector(MissionModel):
141
 
142
  # Smart Selection: If 'world' is requested, use YOLO-World
143
  if self.model_type == "world":
144
- filename = "yolov8s-world.pt"
145
  else:
146
  filename = f"yolov8{self.model_type}.pt"
147
 
 
141
 
142
  # Smart Selection: If 'world' is requested, use YOLO-World
143
  if self.model_type == "world":
144
+ filename = "yolov8n-world.pt"
145
  else:
146
  filename = f"yolov8{self.model_type}.pt"
147