Update processing_videollama3.py

processing_videollama3.py

@@ -304,14 +304,17 @@ class Videollama3Qwen2Processor(ProcessorMixin):
             else:
                 raise ValueError(f"Unsupported image path type: {type(image_path)}")
             return images
-            
-        if isinstance(image_path, list):
-            images = [load_single_image(f) for f in image_path]
-        elif isinstance(image_path, str) and os.path.isdir(image_path):
-            images = [Image.open(os.path.join(image_path, f)).convert('RGB') for f in sorted(os.listdir(image_path))]
-        else:
-            images = [load_single_image(image_path)]
-        return images
+
+        try:
+            if isinstance(image_path, list):
+                images = [load_single_image(f) for f in image_path]
+            elif isinstance(image_path, str) and os.path.isdir(image_path):
+                images = [Image.open(os.path.join(image_path, f)).convert('RGB') for f in sorted(os.listdir(image_path))]
+            else:
+                images = [load_single_image(image_path)]
+            return images
+        except:
+            raise ValueError(f"Error when loading images: {type(image_path)}")
 
     def load_video(
         self,