actual size of those images 178x218, not 224x224. Replace this target_size = (224, 224) by target_size = (178, 218)
target_size = (224, 224)
target_size = (178, 218)