Update ToolSet.py

ToolSet.py

@@ -253,8 +253,9 @@ Do NOT send commands that block indefinitely (e.g., `input()`).""",
 
 
 class YouTubeFrameExtractor:
-    def __init__(self, model_path: str = 'yolov8n.pt', frame_rate: int = 1):
+    def __init__(self, model_path: str, frame_rate: int = 1):
         # Load YOLOv8 model
+        model_path = os.getenv("YOLO_CONFIG_DIR")
         self.model = YOLO(model_path)
         self.frame_rate = frame_rate  # frames per second to sample