code: def mnist_model():
      """Creates a MNIST model."""
      model = tf.keras.models.Sequential()
      model.add(tf.keras.layers.Flatten())
      model.add(tf.keras.layers.Dense(128, activation="relu"))
      model.add(tf.keras.layers.Dense(10, activation="softmax"))
      return model
  
imports: import tensorflow.compat.v2
  from tensorflow.compiler.mlir.tensorflow.tests.tf_saved_model import common
label_desc: Use image node_id 1 for calling a global function with example usage: mnist_model() and returns: model