human
3.1. Implement serving input receiver functions
Serving function
We create serving input function that expects features dictionary and returns the prediction pobabilities from the model"
Please write code following the instructions in jupyter notebook style.
 ... Give me a code to 3?
assistant
#Here is code to 3.

def make_features_serving_fn(model):

  @tf.function
  def serve_features_fn(features):
    return {MODEL_OUTPUT_KEY: model(features)}

  return serve_features_fn