Update main.py

main.py

@@ -418,7 +418,7 @@ def download_npz_file(video_name: str) -> str:
     """
     try:
         # Construct the file path in the dataset repo
-        file_path = f"{HF_DATASET_SUBFOLDER}/{video_name}.npz"
+        file_path = f"{video_name}.npz"
         
         # Check if file already exists in cache
         local_path = os.path.join(CACHE_DIR, f"{video_name}.npz")
@@ -455,7 +455,7 @@ def get_available_videos_from_hf():
         # Filter for .npz files in the I3D subfolder
         videos = []
         for file in files:
-            if file.startswith(f"{HF_DATASET_SUBFOLDER}/") and file.endswith('.npz'):
+            if file.startswith(f"") and file.endswith('.npz'):
                 # Extract the full filename without extension
                 # For files like "I3D/OP02-R02-TurkeySandwich.npz"
                 video_name = os.path.basename(file).replace('.npz', '')