This application uses TensorFlow Serving to serve an image classifier model
To run this application, clone the repo:
git clone https://github.com/risenW/tensorflow_serving_app.git
cd tensorflow_serving_app
pip install -r requirements.txt
Install Tensorflow Serving
Start TF serving server
docker run -p 8501:8501 --name tfserving_classifier \
--mount type=bind,source=tf-server/img_classifier/,target=/models/img_classifier \
-e MODEL_NAME=img_classifier -t tensorflow/serving
To make predictions, send HTTP request to server from another terminal.
python predict.py
Extra:
To retrain the model, run the python model.py