Skip to content
No description, website, or topics provided.
Python
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
.gitignore
LICENSE
README.md
app.py
imagenet_class_index.json
requirements.txt

README.md

PyTorch Flask API

This repo contains a sample code to show how to create a Flask API server by deploying our PyTorch model

If you'd like to learn how to deploy to Heroku, then check this repo.

How to

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

License

The mighty MIT license. Please check LICENSE for more details.

You can’t perform that action at this time.