Skip to content

Latest commit

 

History

History
31 lines (21 loc) · 621 Bytes

README.md

File metadata and controls

31 lines (21 loc) · 621 Bytes

PyTorch with Flask Demo

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