code: def get_keras_model(width=128, activation="relu"):
      """Returns an instance of a Keras Sequential model.
      https://www.tensorflow.org/api_docs/python/tf/keras/Sequential"""
      return tf.keras.models.Sequential(
          [
              tf.keras.layers.Flatten(input_shape=(28, 28)),
              tf.keras.layers.Dense(width, activation=activation),
              tf.keras.layers.Dense(width, activation=activation),
              tf.keras.layers.Dense(10, activation=None),
          ]
      )
  
imports: import warnings
  import tensorflow
  from absl import app, flags
label_desc: Use image node_id 1 for calling a global function with example usage: get_keras_model(width, activation) and returns: tf