I am trying to write a model to classify the handwritten digits from the mnist dataset. I am currently using a model that looks like this: \'\'\' model = tf.keras.models.Seq