Update app.py

app.py

@@ -12,7 +12,8 @@ num_videos = len(bench)
 
 prompts = ds.load_dataset(
     "meta-ai-for-media-research/movie_gen_video_bench",
-    split="test"
+    split="test",
+    streaming=True
 )
 prompts = [p["prompt"] for p in prompts][:num_videos]
 
@@ -35,7 +36,7 @@ with gr.Blocks() as demo:
     )
 
     def set_video(new_prompt):
-        new_id = prompts.find(new_prompt)
+        new_id = prompts.index(new_prompt)
         return gr.Video(
             value=bench[new_id]["video"]["path"],
             loop=True,
@@ -43,7 +44,7 @@ with gr.Blocks() as demo:
         )
 
     def decrement(current_prompt):
-        current_id = prompts.find(current_prompt)
+        current_id = prompts.index(current_prompt)
         return gr.Dropdown(
             label="Video",
             choices=prompts,
@@ -51,7 +52,7 @@ with gr.Blocks() as demo:
         )
 
     def increment(current_prompt):
-        current_id = prompts.find(current_prompt)
+        current_id = prompts.index(current_prompt)
         return gr.Dropdown(
             label="Video",
             choices=prompts,