Spaces:
Sleeping
Sleeping
Update ToolSet.py
Browse files- 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 |
|