The image_dataset_from_directory function returns a tf.data.Dataset object when the labels parameter is set to 'inferred'. In the code, however, the value used for labels is None.Please refer to image_dataset_from_directory