Update app.py

app.py

@@ -4,12 +4,12 @@ import shutil
 shutil.move("repository.py", "/usr/local/lib/python3.10/site-packages/huggingface_hub/repository.py")
 import gradio as gr
 import random
-from huggingface_hub import Repository
+from huggingface_hub import Repository,HfApi
 # from datasets import load_dataset
 from datasets import config
 
 hf_token = os.environ['hf_token']  # 确保环境变量中有你的令牌
-submission_url = "Vchitect/VBench_sampled_video"  # 数据集的 URL
+submission_url =  # 数据集的 URL
 local_dir = "VBench_sampled_video"  # 本地文件夹路径
 # dataset = load_dataset("Vchitect/VBench_sampled_video")
 # print(os.listdir("~/.cache/huggingface/datasets/Vchitect___VBench_sampled_video/"))
@@ -28,10 +28,21 @@ def print_directory_contents(path, indent=0):
     except PermissionError:
         print('    ' * indent + "[权限错误，无法访问该目录]")
 
+# 拉取数据集
+os.makedirs(local_dir, exists_ok=True)
+hf_api = HfApi(token=hf_token)
+repo_id = "Vchitect/VBench_sampled_video"
+dataset_files = api.list_repo_files(repo_id=repo_id, token=hf_token, repo_type='dataset')
 
-# 克隆数据集
-submission_repo = Repository(local_dir=local_dir, clone_from=submission_url, use_auth_token=hf_token, repo_type="dataset")
-submission_repo.git_pull()  # 更新本地仓库
+for file in dataset_files:
+    print(file)
+    api.download_file(repo_id=repo_id, filename=file, token=hf_token,repo_type='dataset',cache_dir=local_dir)
+
+# 下载数据集文件
+for file in dataset_files:
+    api.download_file(repo_id=repo_id, filename=file, token=hf_token)
+
+repo = HfApi(endpoint="https://huggingface.co", token=hf_token)
 
 model_names = os.listdir(local_dir)