Update mission_models.py
Browse files- 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 = "
|
| 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 |
|