This project trains a simple DL model on the MNIST dataset to recognize handwritten digits.
pip install -r requirements.txt
📂 src
├── 📂 checkpoints
│ ├── 📂 checkpoint_1
│ └── 📂 checkpoint_2
├── 📂 data
│ ├── 📂 MNIST
│ └── 📄 dataset.py
├── 📂 models
│ └── 📄 model.py
├── 📂 test_images
│ ├── 📄 00.png
│ ├── 📄 01.png
│ └── ...
├── 📂 training
│ ├── 📄 evaluation.py
│ └── 📄 train.py
├── 📂 utils
│ └── 📄 utils.py
├── 📄 main.py
└── 📄 mnist.ipynb
Train & Eval
python src/main.py
📂 api_app
├── 📂 models
│ └── 📄 model.py
├── 📂 utils
│ ├── 📄 preprocess.py
│ └── 📄 utils.py
├── 📄 api.py
├── 📄 inference.py
└── 📄 run.py
Run FastAPI
python api_app/run.py
📂 streamlit_app
├── 📂 models
│ └── 📄 model.py
├── 📂 utils
│ ├── 📄 predict.py
│ ├── 📄 preprocess.py
│ └── 📄 utils.py
└── 📄 streamlit_app.py
Run streamlit app
streamlit run streamlit_app/streamlit_app.py