Text Column not working with Image Folder

Can you replace data_dir= with data_files= in the last load_dataset call and try again?