feat: add prediction script and model file for MNIST digit classification

feat: update training loop with learning rate scheduler and progress bar enhancements

feat: enhance training loop with tqdm progress bar and configurable parameters

feat: add seed setting for deterministic training

feat: update model training and logging, add mypy cache to .gitignore

feat: baseline model

feat: add init model

chore: add project config

Update README.md

Update README.md

Create README.md

initial commit