Skip to content

McGillAISociety/mnist-flask-demo

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published