diff --git a/contrib/tf_serving/tfs_sample_client.py b/contrib/tf_serving/tfs_sample_client.py index 5208db9d..84291784 100644 --- a/contrib/tf_serving/tfs_sample_client.py +++ b/contrib/tf_serving/tfs_sample_client.py @@ -35,7 +35,7 @@ def get_image_quality_predictions(image_path, model_name): request.model_spec.signature_name = 'image_quality' request.inputs['input_image'].CopyFrom( - tf.contrib.util.make_tensor_proto(np.expand_dims(image, 0)) + tf.compat.v1.make_tensor_proto(np.expand_dims(image, 0)) ) response = stub.Predict(request, 10.0)