Update app.py

app.py

@@ -1,12 +1,29 @@
 import gradio as gr
 import random
-from huggingface_hub import Repository
-# from datasets import load_dataset
+# from huggingface_hub import Repository
+from datasets import load_dataset
 import os
 
-hf_token = os.environ['hf_token']  # 确保环境变量中有你的令牌
-submission_url = "Vchitect/VBench_sampled_video"  # 数据集的 URL
-local_dir = "VBench_sampled_video"  # 本地文件夹路径
+# hf_token = os.environ['hf_token']  # 确保环境变量中有你的令牌
+# submission_url = "Vchitect/VBench_sampled_video"  # 数据集的 URL
+# local_dir = "VBench_sampled_video"  # 本地文件夹路径
+dataset = load_dataset("Vchitect/VBench_sampled_video", use_auth_token=hf_token)
+# print(os.listdir("~/.cache/huggingface/datasets/Vchitect___VBench_sampled_video/"))
+root = "~/.cache/huggingface/datasets/Vchitect___VBench_sampled_video/"
+def print_directory_contents(path, indent=0):
+    # 打印当前目录的内容
+    try:
+        for item in os.listdir(path):
+            item_path = os.path.join(path, item)
+            print('    ' * indent + item)  # 使用缩进打印文件或文件夹
+            if os.path.isdir(item_path):  # 如果是目录，则递归调用
+                print_directory_contents(item_path, indent + 1)
+    except PermissionError:
+        print('    ' * indent + "[权限错误，无法访问该目录]")
+
+# 调用函数，传入你想要打印的目录路径
+print_directory_contents(root)  # 替换为实际路径
+# local_dir = 
 
 # 克隆数据集
 submission_repo = Repository(local_dir=local_dir, clone_from=submission_url, use_auth_token=hf_token, repo_type="dataset")