Update main.py

main.py

@@ -395,7 +395,7 @@ VIS_CONFIG = {
 }
 
 # Cache directory for downloaded .npz files
-CACHE_DIR = Path("./data/cache/I3D")
+CACHE_DIR = Path("./data/I3D")
 CACHE_DIR.mkdir(parents=True, exist_ok=True)
 
 # Hugging Face dataset repository
@@ -418,7 +418,7 @@ def download_npz_file(video_name: str) -> str:
         # Download the .npz file to cache
         downloaded_path = hf_hub_download(
             repo_id=HF_DATASET_REPO,
-            filename=f"{HF_NPZ_SUBFOLDER}/{npz_filename}",
+            filename=f"{npz_filename}",
             repo_type="dataset",
             cache_dir=CACHE_DIR,
             local_dir=CACHE_DIR,
@@ -700,7 +700,6 @@ def get_available_videos():
         repo_files = list_repo_files(
             repo_id=HF_DATASET_REPO,
             repo_type="dataset",
-            subfolder=HF_NPZ_SUBFOLDER
         )
         # Filter for .npz files and extract video names
         videos = [file.replace('.npz', '') for file in repo_files if file.endswith('.npz')]