instead of using
model.save(KERAS_PATH)
or
tf.saved_model.save() #this will cause an error
use:
model.export(SAVEDMODEL_PATH)