MNIST-Demo Run: python main.py Upload a test image in TestImages to predict the digit. The network used is pretrained and stored in mnist_cnn.pt. Train: python train.py Train the MINST network.