Update trainer.py

trainer.py

@@ -74,7 +74,7 @@ model = tf.keras.Sequential([
     tf.keras.layers.MaxPooling2D(2, 2),
     tf.keras.layers.Flatten(),
     tf.keras.layers.Dense(128, activation=tf.nn.relu),
-    tf.keras.layers.Dense(6, activation=tf.nn.softmax)
+    tf.keras.layers.Dense(3, activation=tf.nn.softmax)
 ])
 
 model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['accuracy'])