Skip to content

Commit 3a0db2f

Browse files
committed
Update ServingController.java
1 parent 33e0852 commit 3a0db2f

File tree

1 file changed

+2
-1
lines changed
  • deep_java_library_bring_your_own_container_serving_local_mode/container/src/main/java/com/example/sagemaker/djl/serving

1 file changed

+2
-1
lines changed

deep_java_library_bring_your_own_container_serving_local_mode/container/src/main/java/com/example/sagemaker/djl/serving/ServingController.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ public class ServingController {
3535
private ZooModel model = null;
3636
private Logger logger = LoggerFactory.getLogger(ServingController.class);
3737

38-
@Value("${model.path:/opt/ml/model/build/pytorch_models/resnet18}")
38+
@Value("${model.path:/opt/ml/model/resnet18}")
3939
private String modelPath;
4040

4141
@GetMapping(value = "/ping", produces = "application/json")
@@ -80,6 +80,7 @@ public ResponseEntity<?> invoke_model(@RequestBody String payload) {
8080

8181
private ZooModel init_model() throws ModelNotFoundException, MalformedModelException, IOException {
8282
logger.info("init_model - Start");
83+
logger.info("init_model - modelPath: {}", modelPath);
8384

8485
Translator<Image, Classifications> translator = ImageClassificationTranslator.builder()
8586
.addTransform(new Resize(256))

0 commit comments

Comments
 (0)