chunk_content:  """Apply val_transforms across a batch."""
      example_batch["pixel_values"] = [val_transforms(image.convert("RGB")) for image in example_batch["image"]]
      return example_batch
  Split the dataset into training and validation sets:
  
  
  	Copied
  splits = dataset.train_test_split(test_size=0.1)
  train_ds = splits["train"]
  val_ds = splits["test"]
  Finally, set the transformation functions for the datasets accordingly:
  
  
  	Copied
  train_ds.set_transform(