Update app.py

app.py

@@ -10,11 +10,17 @@ bench = ds.load_dataset(
 bench = bench.cast_column("video", ds.Video(decode=False))
 num_videos = len(bench)
 
+prompts = ds.load_dataset(
+    "meta-ai-for-media-research/movie_gen_video_bench",
+    split="test"
+)
+prompts = [p["prompt"] for p in prompts][:num_videos]
+
 with gr.Blocks() as demo:
     header = gr.Markdown(
         """
-        # Goku MovieGen Bench Video Explorer
-        Use this interface to view videos generated by [Goku](https://saiyan-world.github.io/goku/). For more information, see [Goku: Flow Based Video Generate Models on arXiv](https://arxiv.org/abs/2502.04896).                                       """.strip()
+        # Goku MovieGen Video Bench Video Explorer
+        Use this interface to view videos generated by [Goku](https://saiyan-world.github.io/goku/) using the promps from [Meta AI's MovieGen Video Bench](https://huggingface.co/datasets/meta-ai-for-media-research/movie_gen_video_bench). For more information, see [Goku: Flow Based Video Generate Models on arXiv](https://arxiv.org/abs/2502.04896).                                       """.strip()
     )
     video = gr.Video(
         value=bench[0]["video"]["path"],
@@ -24,36 +30,39 @@ with gr.Blocks() as demo:
 
     dropdown = gr.Dropdown(
         label="Video",
-        choices=range(num_videos),
-        value=0
+        choices=prompts,
+        value=prompts[0]
     )
 
-    def set_video(new_id):
+    def set_video(new_prompt):
+        new_id = prompts.find(new_prompt)
         return gr.Video(
             value=bench[new_id]["video"]["path"],
             loop=True,
             autoplay=True,
         )
 
-    def decrement(current_id):
+    def decrement(current_prompt):
+        current_id = prompts.find(current_prompt)
         return gr.Dropdown(
             label="Video",
-            choices=range(num_videos),
-            value=max(0, current_id-1)
+            choices=prompts,
+            value=prompts[max(0, current_id-1)]
         )
 
-    def increment(current_id):
+    def increment(current_prompt):
+        current_id = prompts.find(current_prompt)
         return gr.Dropdown(
             label="Video",
-            choices=range(num_videos),
-            value=min(num_videos-1, current_id+1)
+            choices=prompts,
+            value=prompts[min(num_videos-1, current_id+1)]
         )
 
     def random_video():
         return gr.Dropdown(
             label="Video",
-            choices=range(num_videos),
-            value=random.randint(0, num_videos-1)
+            choices=prompts,
+            value=prompts[random.randint(0, num_videos-1)]
         )
 
     with gr.Row():