keras_model_path = "/tmp/keras_save" model.save(keras_model_path) restored_keras_model = tf.keras.models.load_model(keras_model_path)