It seems like ImageDataGenerator used in fid is deprecated, could this be updated to the image_dataset_from_directory function?