https://www.bragitoff.com/2022/01/streamlit-app-for-digit-recognition-trained-on-the-mnist-dataset-using-pytorch/