GuangTingZheng

@pokameng I solve this issue: from io import BytesIO from PIL import Image def pre_process(examples): processed = {} processed[image_key] = [Image.open(BytesIO(tform(im['bytes']))) for im in examples[image_column]] processed[caption_key] = examples[text_column] return processed