Update modelutil.py

modelutil.py

@@ -2,11 +2,11 @@ import tensorflow as tf
 
 def create_model():
 
-    LAYERS = [tf.keras.layers.Conv2D(32, (3, 3), activation='relu', input_shape=(28, 28, 1), name="convlayer1"),
-              tf.keras.layers.Conv2D(64, (3, 3), activation='relu', name="convlayer2"),
+    LAYERS = [
               tf.keras.layers.Flatten(input_shape=[28,28], name="inputlayer"),
               tf.keras.layers.Dense(300, activation='relu', name="hiddenlayer1"),
-              tf.keras.layers.Dense(100, activation='relu', name="hiddenlayer2"),
+              tf.keras.layers.Dense(256, activation='relu', name="hiddenlayer1"),
+              tf.keras.layers.Dense(128, activation='relu', name="hiddenlayer2"),
               tf.keras.layers.Dense(10, activation='softmax', name="outputlayer")]
     
     model = tf.keras.models.Sequential(LAYERS)