def METHOD_NAME(input_shape):
    return tf.keras.Sequential(
        [
            tf.keras.layers.Flatten(input_shape=input_shape[1:]),
            tf.keras.layers.Dense(128, activation="relu"),
            tf.keras.layers.Dense(10),
        ]
    )