From 71d593df176fab9c37ad7202d906f7f253c39237 Mon Sep 17 00:00:00 2001 From: punkerpunker Date: Sun, 5 Nov 2023 14:01:14 +0000 Subject: [PATCH] docker image tag --- deployments/tensorflow-serving/README.md | 4 ++-- private_detector/private_detector.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) 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():