Skip to content

Commit

Permalink
fix some typos
Browse files Browse the repository at this point in the history
Signed-off-by: zhlsunshine <huailong.zhang@intel.com>
  • Loading branch information
zhlsunshine committed Mar 21, 2024
1 parent ea62e34 commit e1d75eb
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 11 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -155,7 +155,7 @@ from mlserver import MLModel
from mlserver.utils import get_model_uri
from mlserver.types import InferenceRequest, InferenceResponse, ResponseOutput, Parameters
from mlserver.codecs import DecodedParameterName
from joblib import load # Add this line for importing joblib module
from joblib import load

import logging
import numpy as np
Expand All @@ -177,7 +177,7 @@ class CustomMLModel(MLModel): # pylint:disable=c-extension-no-member
logger.info("Model load URI: {model_uri}")
if exists(model_uri):
logger.info(f"Loading MNIST model from {model_uri}")
self._model = load(model_uri) # Changed 'joblib.load' to 'load'
self._model = load(model_uri)
logger.info("Model loaded successfully")
else:
logger.info(f"Model not exist in {model_uri}")
Expand Down Expand Up @@ -260,7 +260,7 @@ docker build -t <DOCKER-HUB-ORG>/custom-model-server:0.1 .

```
> **Note**: Please use the `--build-arg` to add the HTTP_PROXY if necessary, such as:
> **Note**: Please use the `--build-arg` to add the http proxy if there is proxy in user's environment, such as:
```shell
docker build --build-arg HTTP_PROXY=http://<DOMAIN-OR-IP>:PORT --build-arg HTTPS_PROXY=http://<DOMAIN-OR-IP>:PORT -t <DOCKER-HUB-ORG>/custom-model-server:0.1 .
```
Expand Down Expand Up @@ -385,12 +385,8 @@ curl -s -X POST -k "http://localhost:8008/v2/models/${MODEL_NAME}/infer" -d '{"i
{
"name": "predict",
"datatype": "INT64",
"shape": [
1
],
"data": [
8
]
"shape": [1],
"data": [8]
}
]
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
from mlserver.utils import get_model_uri
from mlserver.types import InferenceRequest, InferenceResponse, ResponseOutput, Parameters
from mlserver.codecs import DecodedParameterName
from joblib import load # Add this line for importing joblib module
from joblib import load

import logging
import numpy as np
Expand All @@ -27,7 +27,7 @@ async def load(self) -> bool:
logger.info("Model load URI: {model_uri}")
if exists(model_uri):
logger.info(f"Loading MNIST model from {model_uri}")
self._model = load(model_uri) # Changed 'joblib.load' to 'load'
self._model = load(model_uri)
logger.info("Model loaded successfully")
else:
logger.info(f"Model not exist in {model_uri}")
Expand Down

0 comments on commit e1d75eb

Please sign in to comment.