Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import random
 # from huggingface_hub import Repository
 from datasets import load_dataset
+from datasets import config
 import os
 
 # hf_token = os.environ['hf_token']  # 确保环境变量中有你的令牌
@@ -10,7 +11,7 @@ import os
 dataset = load_dataset("Vchitect/VBench_sampled_video")
 # print(os.listdir("~/.cache/huggingface/datasets/Vchitect___VBench_sampled_video/"))
 # root = "~/.cache/huggingface/datasets/Vchitect___VBench_sampled_video/"
-root = os.environ['HF_DATASETS_CACHE']
+print(config.HF_DATASETS_CACHE)
 print(root)
 def print_directory_contents(path, indent=0):
     # 打印当前目录的内容