Fix keras model saving error into TF SavedModel format.
Fix keras model saving error into TF SavedModel format.

Now that keras 3 doesn't support model saving format TF SavedModel directly via `model.save()`, we need to use `tf.saved_model.save()` to save the model.
