Update app.py

app.py

@@ -10,7 +10,9 @@ from datasets import config
 
 hf_token = os.environ['hf_token']  # 确保环境变量中有你的令牌
 
-local_dir = "VBench_sampled_video"  # 本地文件夹路径
+local_dir = "VBench-I2V_sampled_video"  # 本地文件夹路径
+
+
 # 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/"
@@ -33,8 +35,10 @@ 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"
+# repo_id = "Vchitect/VBench-I2V_sampled_video"
 
 model_names=[]
+# for i in hf_api.list_repo_tree('Vchitect/VBench-I2V_sampled_video',repo_type='dataset'):
 for i in hf_api.list_repo_tree('Vchitect/VBench_sampled_video',repo_type='dataset'):
     model_name = i.path
     if '.git' not in model_name and '.md' not in model_name:
@@ -52,6 +56,8 @@ with open("videos_by_dimension.json") as f:
     # all_videos = json.load(f)
 
 types = ['appearance_style', 'color', 'temporal_style', 'spatial_relationship', 'temporal_flickering', 'scene', 'multiple_objects', 'object_class', 'human_action', 'overall_consistency', 'subject_consistency']
+# types = ['subject_consistency', 'background_consistency', 'motion_smoothness', 'dynamic_degree', 'aesthetic_quality', 'imaging_quality', 'i2v_subject', 'i2v_background', 'camera_motion']
+
 
 def get_video_path_local(model_name, type, prompt):
     if 'Show-1' in model_name:
@@ -183,7 +189,7 @@ def show_feedback_button():
 with gr.Blocks() as interface:
     gr.Markdown("# VBench Video Arena")
     gr.Markdown("""
-**VBench Video Arena: Watch AI-Generated Videos Instantly** (powered by [VBench](https://github.com/Vchitect/VBench) and [VBench Leaderboard](https://huggingface.co/spaces/Vchitect/VBench_Leaderboard))\n
+**VBench-I2V Video Arena: Watch AI-Generated Videos Instantly** (powered by [VBench](https://github.com/Vchitect/VBench) and [VBench Leaderboard](https://huggingface.co/spaces/Vchitect/VBench_Leaderboard))\n
 - **Random 2 Videos**: Randomly selects two models to compare on the same ability dimension and text prompt.\n
 - **Play Selection** Allows users to choose a model, dimension, and text prompt from drop-down menus and view the corresponding videos. """)