Update run.py

run.py

@@ -35,9 +35,18 @@ def fast_caption(sys_prompt, usr_prompt, temp, top_p, max_tokens, model, key, en
         video_paths = video_paths["train"]
         # Process all videos in the dataset
         all_captions = []
-        for video_path in video_paths:
+        for video_path_url in video_paths:
             print("video_path")
-            video_path = video_path["id"]
+            video_path_url = video_path_url["id"]
+            # 使用requests下载文件到临时文件
+            response = requests.get(video_url, stream=True)
+            if response.status_code == 200:
+                with tempfile.NamedTemporaryFile(delete=False, suffix=".mp4") as temp_video_file:
+                    temp_video_file.write(response.content)
+                    video_path = temp_video_file.name
+            else:
+                raise Exception(f"Failed to download video, status code: {response.status_code}")
+                
             if video_path.endswith('.mp4'):  # 假设我们只处理.mp4文件
                 processor = VideoProcessor(frame_format=frame_format, frame_limit=frame_limit)
                 frames = processor._decode(video_path)