Update image_processing_paddleocr_vl.py

image_processing_paddleocr_vl.py

@@ -338,10 +338,12 @@ class PaddleOCRVLImageProcessor(BaseImageProcessor):
         """
         images = make_list_of_images(images)
 
+        if input_data_format is None:
+            input_data_format = ChannelDimension.LAST if isinstance(images[0], Image.Image) else infer_channel_dimension_format(images[0])
+
         if do_convert_rgb:
             images = [convert_to_rgb(image) for image in images]
 
-        # All transformations expect numpy arrays.
         images = [to_numpy_array(image) for image in images]
 
         if is_scaled_image(images[0]) and do_rescale:
@@ -349,14 +351,10 @@ class PaddleOCRVLImageProcessor(BaseImageProcessor):
                 "It looks like you are trying to rescale already rescaled images. If the input"
                 " images have pixel values between 0 and 1, set `do_rescale=False` to avoid rescaling them again."
             )
-        if input_data_format is None:
-            # We assume that all images have the same channel dimension format.
-            input_data_format = infer_channel_dimension_format(images[0])
 
         height, width = get_image_size(images[0], channel_dim=input_data_format)
         resized_height, resized_width = height, width
         processed_images = []
-
         for image in images:
             if do_resize:
                 resized_height, resized_width = smart_resize(