ynhe commited on
Commit
7fc7ae9
·
verified ·
1 Parent(s): 1dfa4f0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,10 +1,11 @@
1
  import gradio as gr
2
  import random
3
  from datasets import load_dataset
 
 
4
 
5
  # 加载数据集
6
- dataset = load_dataset("Vchitect/VBench_sampled_video")
7
-
8
  # 随机选择一个视频
9
  def get_random_video():
10
  # 随机选择一个索引
 
1
  import gradio as gr
2
  import random
3
  from datasets import load_dataset
4
+ import os
5
+ hf_token = os.environ['hf_token']
6
 
7
  # 加载数据集
8
+ dataset = load_dataset("Vchitect/VBench_sampled_video", use_auth_token=hf_token)
 
9
  # 随机选择一个视频
10
  def get_random_video():
11
  # 随机选择一个索引