notsakeeb commited on
Commit
48d8e67
·
verified ·
1 Parent(s): 27ff947

Update ToolSet.py

Browse files
Files changed (1) hide show
  1. ToolSet.py +1 -2
ToolSet.py CHANGED
@@ -253,12 +253,11 @@ Do NOT send commands that block indefinitely (e.g., `input()`).""",
253
 
254
 
255
  class YouTubeFrameExtractor:
256
- def __init__(self, model_path: str, frame_rate: int = 1):
257
  # Load YOLOv8 model
258
  YOLO_PATH_CONFIG = Path("yolo")
259
  YOLO_PATH_CONFIG.mkdir(exist_ok=True)
260
  settings.update({"runs_dir": YOLO_PATH_CONFIG})
261
- model_path = YOLO_PATH_CONFIG
262
  self.model = YOLO(model_path)
263
  self.frame_rate = frame_rate # frames per second to sample
264
 
 
253
 
254
 
255
  class YouTubeFrameExtractor:
256
+ def __init__(self, model_path: str = 'yolov8n.pt', frame_rate: int = 1):
257
  # Load YOLOv8 model
258
  YOLO_PATH_CONFIG = Path("yolo")
259
  YOLO_PATH_CONFIG.mkdir(exist_ok=True)
260
  settings.update({"runs_dir": YOLO_PATH_CONFIG})
 
261
  self.model = YOLO(model_path)
262
  self.frame_rate = frame_rate # frames per second to sample
263