Django + PyTorch Image Classification App

A simple Django web app classifying uploaded images using a pretrained PyTorch DenseNet.

As an example of how to integrate PyTorch with Django and deploy it on Heroku.

• Live demo: Heroku (might take multiple seconds to start the free Heroku instance)
• Corresponding blog post: Using PyTorch Inside a Django App

Setup

pip install -r requirements

Usage

Live demo: Heroku (might take multiple seconds to start the free Heroku instance)

Development

python manage.py runserver

The app is running on http://localhost:8000/

Production Deployment on Heroku

See description in blog post.