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