Update videollama2/mm_utils.py

videollama2/mm_utils.py

@@ -4,6 +4,7 @@ import math
 import base64
 import traceback
 from io import BytesIO
+import gdown
 
 import cv2
 import torch
@@ -131,6 +132,10 @@ def frame_sample(duration, mode='uniform', num_frames=None, fps=None):
 
 
 def process_video(video_path, processor, s=None, e=None, aspect_ratio='pad', num_frames=NUM_FRAMES):
+    output = 'Temp.mp4'
+    gdown.download(video_path, output, quiet=False)
+    video_path = 'Temp.mp4'
+    logging.info(f"video downloaded form: {url}")
     if isinstance(video_path, str):
         if s is not None and e is not None:
             s = s if s >= 0. else 0.