---
title: Digits
emoji: 🔢
colorFrom: indigo
colorTo: indigo
sdk: gradio
sdk_version: 3.12.0
app_file: app.py
pinned: false
license: apache-2.0
---

Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

This gradio app predicts digits using a convolutive neural network (CNN) that was trained on the MNIST hand-drawn digit data set:
@article{lecun2010mnist,
  title={MNIST handwritten digit database},
  author={LeCun, Yann and Cortes, Corinna and Burges, CJ},
  journal={ATT Labs [Online]. Available: http://yann.lecun.com/exdb/mnist/},
  volume={2},
  year={2010}
}

The PyTorch network architecture:
Sequential(
  (conv1): Conv2d(1, 32, kernel_size=(5, 5), stride=(1, 1), padding=(2, 2))
  (relu1): ReLU()
  (pool1): MaxPool2d(kernel_size=2, stride=2, padding=0, dilation=1, ceil_mode=False)
  (conv2): Conv2d(32, 64, kernel_size=(5, 5), stride=(1, 1), padding=(2, 2))
  (relu2): ReLU()
  (pool2): MaxPool2d(kernel_size=2, stride=2, padding=0, dilation=1, ceil_mode=False)
  (flatten): Flatten(start_dim=1, end_dim=-1)
  (fc1): Linear(in_features=3136, out_features=1024, bias=True)
  (dropout): Dropout(p=0.5, inplace=False)
  (fc2): Linear(in_features=1024, out_features=10, bias=True)
)

The model was trained using cross entropy loss function, the Adam stochastic optimizer.
Training was done for 10 epochs, using batch size of 64, and with a learning rate of 0.001.

The training and validation accuracy after each epoch was as follows:

 Epoch 1 ---- train accuracy: 0.9607 ---- val accuracy: 0.9885
 Epoch 2 ---- train accuracy: 0.9870 ---- val accuracy: 0.9894
 Epoch 3 ---- train accuracy: 0.9893 ---- val accuracy: 0.9892
 Epoch 4 ---- train accuracy: 0.9925 ---- val accuracy: 0.9914
 Epoch 5 ---- train accuracy: 0.9932 ---- val accuracy: 0.9872
 Epoch 6 ---- train accuracy: 0.9947 ---- val accuracy: 0.9904
 Epoch 7 ---- train accuracy: 0.9946 ---- val accuracy: 0.9888
 Epoch 8 ---- train accuracy: 0.9947 ---- val accuracy: 0.9920
 Epoch 9 ---- train accuracy: 0.9962 ---- val accuracy: 0.9912
 Epoch 10 ---- train accuracy: 0.9960 ---- val accuracy: 0.9915