The predict client is meant to be used with a model served by TensorFlow Serving. Because tfserving model server runs a gRPC service, it cannot be requested by just sending a normal HTTP request. The predict_client package is a grcp client that can request the service.
Feel free to use this package to integrate your python apis with tfserving models. Or clone the repo and make your own client.
Read my blog posts about TensorFlow Serving:
pip install git+ssh://[email protected]/epigramai/tfserving-python-predict-client.git
There is one here https://hub.docker.com/r/epigramai/model-server/
Check out the examples. The download scripts with test data and models will not work unless you have access to Epigram AI GCS buckets, but feel free to use the examples as a starting point.
Although the models and test data is hidden to the public, the predict client is open source.
def init(self, host, model_name, model_version):
- host: the host (e.g. localhost:9000)
- model_name: your model name, e.g. 'mnist'
- model_version: model version, e.g. 1.
ProdClient.predict(self, request_data, request_timeout=10):
- request_data: A list of input tensors, see the example.
- request_timeout: timeout sent to the grcp stub
from predict_client.prod_client import ProdClient
client = ProdClient('localhost:9000', 'mnist', 1)
client.predict(request_data)
The predict function returns a dictionary with keys and values for each output tensor. The values in the dictionary will have the same shapes as the output tensor's shape. If an error occurs, predict will return an empty dict.
def init(self, model_path):
- model_path
InMemoryClient.predict(self, request_data, request_timeout=None):
- request_data and request_timeout same as ProdClient, except request_timeout not used in this client.
from predict_client.inmemory_client import InMemoryClient
client = InMemoryClient('path/to/model.pb')
client.predict(request_data)
The predict function returns a dictionary with keys and values for each output tensor. The values in the dictionary will have the same shapes as the output tensor's shape. If an error occurs, predict will return an empty dict.
def init(self, mock_response):
- mock_response
MockClient.predict(self, request_data, request_timeout=None):
- request_data and request_timeout same as ProdClient, except request_timeout not used in mock client.
from predict_client.mock_client import MockClient
client = MockClient(mock_response)
client.predict(request_data)
The mock client predict function simply returns the mock response.
pip install grpcio-tools
python -m grpc_tools.protoc -I protos/ --python_out=predict_client/pbs --grpc_python_out=predict_client/pbs protos/*