Update run.py

run.py

@@ -32,10 +32,12 @@ def fast_caption(sys_prompt, usr_prompt, temp, top_p, max_tokens, model, key, en
         print("begin video_hf")
         # Handle Hugging Face dataset
         video_paths = load_hf_dataset(video_hf, video_hf_auth)
+        video_paths = video_paths["train"]
         # Process all videos in the dataset
         all_captions = []
         for video_path in video_paths:
             print("video_path")
+            video_path = video_path["id"]
             if video_path.endswith('.mp4'):  # 假设我们只处理.mp4文件
                 processor = VideoProcessor(frame_format=frame_format, frame_limit=frame_limit)
                 frames = processor._decode(video_path)