Skip to content

goutham-nekkalapu/deploying_machine_learning_models

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

About:

This repository gives ready made template to deploy deep learning model using flask.

How to use it:

  1. Clone this repository :
    git clone https://github.com/goutham-nekkalapu/deploying_ML_models

  2. Install the requirements :
    pip install -r requirements.txt

    If you want to have different versions of python and other deep learning libraries, you can use Anaconda to create environments.

  3. Start the server:
    python flask_server.py

  4. To use the model deployed, run the client program :
    python flask_client.py

Example :

For the image

image

The response from server, has the top-5 predictions, as below:

{ "1": [ "hotdog", "0.491872" ], "2": [ "bagel", "0.139255" ], "3": [ "french loaf", "0.107473" ], "4": [ "pretzel", "0.0680968" ], "5": [ "plate", "0.0374931" ] }

Model used:

The deep learning model deployed here is ResNet-50 and keras library V2.0 is used, this can be modified accordingly to your needs.

About

Flask web server to deploy deep learning, machine learning models

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages