Update src/saving_utils.py

src/saving_utils.py

@@ -37,11 +37,12 @@ def download_files_from_hub(benchmark_types, repo_id="mgyigit/probe-data", repo_
         
         try:
             # Download the file from the specified repo
-            api.download_file(
+            api.hf_hub_download(
                 repo_id=repo_id,
-                path_in_repo=file_name,
-                local_dir="/tmp",
                 repo_type=repo_type,
+                filename=file_name,
+                local_dir="/tmp",
+                token=os.getenv("api-key"),
             )
             print(f"Downloaded {file_name} from {repo_id} to {local_path}")