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

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +28 -0
app.py ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ # 随机选择一个索引
11
+ random_index = random.randint(0, len(dataset['train']) - 1)
12
+ # 获取视频路径
13
+ video_path = dataset['train'][random_index]['video_path']
14
+ return video_path
15
+
16
+ # Gradio 接口
17
+ def display_video():
18
+ video_path = get_random_video()
19
+ return video_path
20
+
21
+ interface = gr.Interface(fn=display_video,
22
+ outputs=gr.Video(label="随机视频展示"),
23
+ inputs=[],
24
+ title="随机视频展示",
25
+ description="从 Vchitect/VBench_sampled_video 数据集中随机展示一个视频。")
26
+
27
+ if __name__ == "__main__":
28
+ interface.launch()