Update processing_videollama3.py

processing_videollama3.py

@@ -9,6 +9,7 @@ from collections import defaultdict
 from typing import Any, List, Union, Dict, Optional, Tuple, TypedDict
 
 import cv2
+import requests
 import ffmpeg
 import imageio
 import json
@@ -298,6 +299,8 @@ class Videollama3Qwen2Processor(ProcessorMixin):
         elif isinstance(image_path, str) and os.path.isdir(image_path):
             # images = [cv2.cvtColor(cv2.imread(os.path.join(image_path, f)), cv2.COLOR_BGR2RGB) for f in sorted(os.listdir(image_path))]
             images = [Image.open(os.path.join(image_path, f)).convert('RGB') for f in sorted(os.listdir(image_path))]
+        elif isinstance(image_path, str) and image_path.startswith("http://") or image_path.startswith("https://"):
+            images = [Image.open(requests.get(image, stream=True).raw)]
         elif isinstance(image_path, list) and isinstance(image_path[0], str):
             # images = [cv2.cvtColor(cv2.imread(f), cv2.COLOR_BGR2RGB) for f in image_path]
             images = [Image.open(f).convert('RGB') for f in image_path]