Update app.py

app.py

@@ -5,6 +5,7 @@ shutil.move("repository.py", "/usr/local/lib/python3.10/site-packages/huggingfac
 import gradio as gr
 import random
 from huggingface_hub import Repository,HfApi
+from huggingface_hub import snapshot_download
 # from datasets import load_dataset
 from datasets import config
 
@@ -33,12 +34,16 @@ os.makedirs(local_dir, exist_ok=True)
 hf_api = HfApi(endpoint="https://huggingface.co", token=hf_token)
 # hf_api = HfApi(token=hf_token)
 repo_id = "Vchitect/VBench_sampled_video"
-dataset_files = hf_api.list_repo_files(repo_id=repo_id, token=hf_token, repo_type='dataset')
 
-for file in dataset_files:
-    print(file)
-    hf_api.download_file(repo_id=repo_id, filename=file, token=hf_token,repo_type='dataset',cache_dir=local_dir)
+downloaded_files = snapshot_download(
+    repo_id=repo_id,
+    token=hf_token,                # 可选：如果你的仓库是私有的
+    repo_type='dataset',           # 如果下载的是数据集而不是模型
+    cache_dir=local_dir,           # 指定缓存/下载位置
+    ignore_patterns=["*.md", "*.txt"]  # 可选：忽略某些模式的文件
+)
 
+print(f"Downloaded files: {downloaded_files}")
 
 model_names = os.listdir(local_dir)