Update ToolSet.py

ToolSet.py

@@ -255,7 +255,10 @@ Do NOT send commands that block indefinitely (e.g., `input()`).""",
 class YouTubeFrameExtractor:
     def __init__(self, model_path: str, frame_rate: int = 1):
         # Load YOLOv8 model
-        model_path = os.getenv("YOLO_CONFIG_DIR")
+        YOLO_PATH_CONFIG = Path("yolo")
+        YOLO_PATH_CONFIG.mkdir(exist_ok=True)
+        settings.update({"runs_dir": YOLO_PATH_CONFIG})
+        model_path = YOLO_PATH_CONFIG
         self.model = YOLO(model_path)
         self.frame_rate = frame_rate  # frames per second to sample