如何在tensorflow模型服务中保存用于服务的tensorflow集线器模型

import tensorflow as tf
from tensorflow import keras

# 构建模型
model = keras.Sequential([
    keras.layers.Dense(64, activation='relu', input_shape=(784,)),
    keras.layers.Dense(10, activation='softmax')
])

# 编译模型
model.compile(optimizer='adam',
              loss='sparse_categorical_crossentropy',
              metrics=['accuracy'])

# 训练模型
model.fit(x_train, y_train, epochs=10, batch_size=32)

# 定义保存路径
export_path = './saved_model'

# 保存模型
tf.saved_model.save(model, export_path)

# 加载模型
loaded_model = tf.saved_model.load(export_path)

# 创建模型服务
model_server = tf.keras.models.load_model(export_path)

# 定义模型服务函数
@tf.function(input_signature=[tf.TensorSpec(shape=[None, 784], dtype=tf.float32)])
def serving_fn(input_data):
    return {'output': model_server(input_data)}

# 保存模型服务
tf.saved_model.save(model_server, export_path, signatures={'serving_default': serving_fn})