Convert Keras Model to CoreML

keras_to_coreml.py

import coremltools

class_labels = ['airplane', 'automobile' ,'bird ','cat ','deer ','dog ','frog ','horse ','ship ','truck']

// model is just a trained keras model.
coreml_model = coremltools.converters.keras.convert(model,
                                                    input_names=['image'],
                                                    image_input_names='image',
                                                    class_labels=class_labels)
coreml_model.save('CIFAR.mlmodel')