Then use 🤗 Datasets[~datasets.Dataset.set_transform] to apply the transforms on the fly:

dataset.set_transform(transforms)

Now when you access the image, you'll notice the image processor has added pixel_values.