def METHOD_NAME():
  '''Create a simple model with two fully connected(fc) layers'''
  model = tf.keras.models.Sequential([
      tf.keras.layers.InputLayer(input_shape=(28, 28)),
      tf.keras.layers.Flatten(),
      tf.keras.layers.Dense(50, activation=tf.nn.relu),
      tf.keras.layers.Dense(10, activation=tf.nn.softmax, name="output")
  ])
  fixed_input = tf.keras.layers.Input(shape=[28, 28],
                                      batch_size=1,
                                      dtype=model.inputs[0].dtype,
                                      name="fixed_input")
  fixed_output = model(fixed_input)
  return tf.keras.models.Model(fixed_input, fixed_output)