Update ToolSet.py

ToolSet.py

@@ -6,7 +6,7 @@ from langchain_experimental.utilities import PythonREPL
 import cv2
 from pathlib import Path
 from yt_dlp import YoutubeDL
-from ultralytics import YOLO
+from ultralytics import YOLO, settings
 from typing import List, Dict
 from typing import TypedDict, Annotated
 from langchain_community.tools.tavily_search import TavilySearchResults
@@ -16,10 +16,6 @@ from langchain.tools import Tool, tool
 from utils import GaiaClient
 
 
-DOWNLOAD_PATH = Path("questions")
-DOWNLOAD_PATH.mkdir(exist_ok=True)
-
-
 @tool
 def download_file(task_id: str, filename: str) -> str:
     """
@@ -28,6 +24,8 @@ def download_file(task_id: str, filename: str) -> str:
         task_id: The id of the task to download a related file.
         filename: Path to store the file to.
     """
+    DOWNLOAD_PATH = Path("questions")
+    DOWNLOAD_PATH.mkdir(exist_ok=True)
     client = GaiaClient(DOWNLOAD_PATH, None, None, api_url="https://agents-course-unit4-scoring.hf.space")
     result = client.download_file_for_task(task_id, filename)
 
@@ -257,6 +255,7 @@ Do NOT send commands that block indefinitely (e.g., `input()`).""",
 class YouTubeFrameExtractor:
     def __init__(self, model_path: str = 'yolov8n.pt', frame_rate: int = 1):
         # Load YOLOv8 model
+        settings.update({"runs_dir": "/path/to/runs"})
         self.model = YOLO(model_path)
         self.frame_rate = frame_rate  # frames per second to sample