MNIST classifier model