Skip to content
No description, website, or topics provided.
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Type Name Latest commit message Commit time
Failed to load latest commit information.

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 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.

You can’t perform that action at this time.