Skip to content

Practice repo of deploying an ML model with Flask and Docker

Notifications You must be signed in to change notification settings

andra-pumnea/ml-with-flask-and-docker

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

ML model with FLask app and Docker

A model that I previously trained to detect whether a comment is hate speech or not is loaded in a Flask webapp and wrapped in a Docker container. This is a simple feed-forward network built with keras, that uses tf-idf to encode the text and achieved 0.66 F1 score during training.

To run the docker image you need to clone the repo and then:

cd ml-with-flask-and-docker
docker build --tag flask-app:1.0 .
docker run --publish 5000:5000 --detach --name app flask-app:1.0
docker start app

The web app will be running at http://localhost:5000/

About

Practice repo of deploying an ML model with Flask and Docker

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published