Just noting that your MNIST demo on https://www.gradio.app/ seems to have a bug.
If you create a small zero it is classified as a 9. Or perhaps it is overly sensitive to line width? Drawing large circles counterclockwise with the small pen seems to be misclassified too.