How to make dataset for instructpix2pix

hello

i want to make dataset from my own image
like this

input_image
edit_prompt
edited_image

i try this code but its only make 2 coulmin (image and text) and i can’t change coulmin name

from datasets import load_dataset

dataset = load_dataset(“imagefolder”, data_dir=“/path/to/pokemon”)