A simple, bare-bones example on how to serve an ML model through a web application, using PyTorch and Flask. Based off the official classifier tutorial.
Images for testing can be found in the images/
directory.
Installing dependencies
$ pip install -r requirements.txt
Training the model
$ python ./model.py train
Testing the model
$ python ./model.py test
Running the server
$ python ./app.py