Update processing_videollama3.py
Browse files
processing_videollama3.py
CHANGED
@@ -295,6 +295,8 @@ class Videollama3Qwen2Processor(ProcessorMixin):
|
|
295 |
|
296 |
def load_images(self, image_path: Union[str, List[str], Image.Image, List[Image.Image]]):
|
297 |
def load_single_image(image_path):
|
|
|
|
|
298 |
if isinstance(image_path, str) and os.path.isfile(image_path):
|
299 |
# images = [cv2.cvtColor(cv2.imread(image_path), cv2.COLOR_BGR2RGB)]
|
300 |
images = Image.open(image_path).convert('RGB')
|
|
|
295 |
|
296 |
def load_images(self, image_path: Union[str, List[str], Image.Image, List[Image.Image]]):
|
297 |
def load_single_image(image_path):
|
298 |
+
if isinstance(image_path, str) and image_path.startswith("file://"):
|
299 |
+
image_path = image_path.replace("file://", "")
|
300 |
if isinstance(image_path, str) and os.path.isfile(image_path):
|
301 |
# images = [cv2.cvtColor(cv2.imread(image_path), cv2.COLOR_BGR2RGB)]
|
302 |
images = Image.open(image_path).convert('RGB')
|