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

Update ToolSet.py

Browse files
Files changed (1) hide show
  1. ToolSet.py +1 -1
ToolSet.py CHANGED
@@ -257,7 +257,7 @@ class YouTubeFrameExtractor:
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
 
 
257
  # Load YOLOv8 model
258
  YOLO_PATH_CONFIG = Path("yolo")
259
  YOLO_PATH_CONFIG.mkdir(exist_ok=True)
260
+ settings.update({"runs_dir": str(YOLO_PATH_CONFIG)})
261
  self.model = YOLO(model_path)
262
  self.frame_rate = frame_rate # frames per second to sample
263