PyTorch Flask API
This repo contains a sample code to show how to create a Flask API server by deploying our PyTorch model. This is a sample code which goes with tutorial.
If you'd like to learn how to deploy to Heroku, then check this repo.
Install the dependencies:
pip install -r requirements.txt
Run the Flask server:
FLASK_ENV=development FLASK_APP=app.py flask run
From another tab, send the image file in a request:
curl -X POST -F file=@cat_pic.jpeg http://localhost:5000/predict
The mighty MIT license. Please check
LICENSE for more details.