mnist serving example using pytorch 2.0 & flask
pip3 install -r requirements.txt
Training mnist with simple network and save a model
python3 train.py
serving model using flask
python3 flask_server.py #if you want to execute at background: nohup python3 flask_server.py &
You can test rest api.
python3 flask_test.py
https://seokhyun2.tistory.com/43