Update app.py

app.py

@@ -81,6 +81,11 @@ class RemoteDatasetLoader:
         video_paths = []
         chunks_size = dataset_info.get("chunks_size", 1000)
 
+        # 创建repo特定的子目录
+        repo_name = self.repo_id.replace('/', '_')  # 将 / 替换为 _ 避免路径问题
+        repo_dir = os.path.join(download_dir, repo_name)
+        os.makedirs(repo_dir, exist_ok=True)
+
         for i, video_key in enumerate(video_keys):
             video_url = self.base_url + dataset_info["video_path"].format(
                 episode_chunk=episode_id // chunks_size,
@@ -88,7 +93,7 @@ class RemoteDatasetLoader:
                 episode_index=episode_id
             )
             video_filename = f"episode_{episode_id}_{video_key}.mp4"
-            local_path = os.path.join(download_dir, video_filename)
+            local_path = os.path.join(repo_dir, video_filename)
             # 优先加载本地有效mp4
             if self._is_valid_mp4(local_path):
                 print(f"Local valid video found: {local_path}")