https://github.com/risenw/tensorflow_serving_app
This application uses TensorFlow Serving to serve an image classifier model
https://github.com/risenw/tensorflow_serving_app
Last synced: 8 months ago
JSON representation
This application uses TensorFlow Serving to serve an image classifier model
- Host: GitHub
- URL: https://github.com/risenw/tensorflow_serving_app
- Owner: risenW
- Created: 2020-09-23T09:25:23.000Z (over 5 years ago)
- Default Branch: master
- Last Pushed: 2020-09-23T09:41:57.000Z (over 5 years ago)
- Last Synced: 2025-03-31T12:57:26.678Z (10 months ago)
- Language: Python
- Size: 1.27 MB
- Stars: 8
- Watchers: 1
- Forks: 2
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
## This application uses [TensorFlow Serving](https://www.tensorflow.org/tfx/guide/serving) to serve an image classifier model
-----------
To run this application, clone the repo:
```bash
git clone https://github.com/risenW/tensorflow_serving_app.git
cd tensorflow_serving_app
pip install -r requirements.txt
```
Install Tensorflow Serving
- [Link](https://www.tensorflow.org/tfx/serving/setup)
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`