diff --git a/deployments/tensorflow-serving/README.md b/deployments/tensorflow-serving/README.md index d4f4e22..f01689f 100644 --- a/deployments/tensorflow-serving/README.md +++ b/deployments/tensorflow-serving/README.md @@ -3,13 +3,13 @@ Tensorflow serving image could be pulled via: ```shell -docker pull +docker pull vazhega/private-detector:0.2.0 ``` Run: ```shell -docker run -p 8501:8501 -e MODEL_NAME=private_detector -t +docker run -p 8501:8501 -e MODEL_NAME=private_detector -t vazhega/private-detector:0.2.0 ``` Ports exposed: diff --git a/private_detector/private_detector.py b/private_detector/private_detector.py index 385f393..50c2a3d 100644 --- a/private_detector/private_detector.py +++ b/private_detector/private_detector.py @@ -12,7 +12,7 @@ from .utils import loss from .utils.efficientnet_config import EfficientNetV2Config from .utils.effnetv2_model import EffNetV2Model from .utils.tensorboard_callback import Callback -from .utils.preprocess import pad_resize_image, preprocess_for_evaluation +from .utils.preprocess import preprocess_for_evaluation class PrivateDetector():