Skip to content

Commit 8b455ed

Browse files
committed
Update examples
1 parent c1b1bb6 commit 8b455ed

File tree

2 files changed

+8
-8
lines changed

2 files changed

+8
-8
lines changed

examples/mnist.java

+4-4
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
import java.nio.file.Files;
88
import java.nio.file.Path;
99

10-
import com.github.tadayosi.torchserve.client.impl.DefaultInference;
10+
import com.github.tadayosi.torchserve.client.TorchServeClient;
1111
import com.github.tadayosi.torchserve.client.model.ApiException;
1212

1313
public class mnist {
@@ -18,10 +18,10 @@ public static void main(String... args) throws Exception {
1818
var zero = Files.readAllBytes(Path.of("src/test/resources/data/0.png"));
1919
var one = Files.readAllBytes(Path.of("src/test/resources/data/1.png"));
2020
try {
21-
var inference = new DefaultInference();
22-
var result0 = inference.predictions(MNIST_MODEL, zero);
21+
var client = TorchServeClient.newInstance();
22+
var result0 = client.inference().predictions(MNIST_MODEL, zero);
2323
System.out.println("Answer> " + result0);
24-
var result1 = inference.predictions(MNIST_MODEL, one);
24+
var result1 = client.inference().predictions(MNIST_MODEL, one);
2525
System.out.println("Answer> " + result1);
2626
} catch (ApiException e) {
2727
e.printStackTrace();

examples/register_mnist.java

+4-4
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
//REPOS mavencentral,jitpack=https://jitpack.io
55
//DEPS com.github.tadayosi:torchserve-client-java:main-SNAPSHOT
66

7-
import com.github.tadayosi.torchserve.client.impl.DefaultManagement;
7+
import com.github.tadayosi.torchserve.client.TorchServeClient;
88
import com.github.tadayosi.torchserve.client.model.ApiException;
99
import com.github.tadayosi.torchserve.client.model.RegisterModelOptions;
1010
import com.github.tadayosi.torchserve.client.model.SetAutoScaleOptions;
@@ -16,10 +16,10 @@ public class register_mnist {
1616

1717
public static void main(String... args) throws Exception {
1818
try {
19-
var management = new DefaultManagement();
20-
var response = management.registerModel(MNIST_URL, RegisterModelOptions.empty());
19+
var client = TorchServeClient.newInstance();
20+
var response = client.management().registerModel(MNIST_URL, RegisterModelOptions.empty());
2121
System.out.println("registerModel> " + response.getStatus());
22-
response = management.setAutoScale(MNIST_MODEL, SetAutoScaleOptions.builder()
22+
response = client.management().setAutoScale(MNIST_MODEL, SetAutoScaleOptions.builder()
2323
.minWorker(1)
2424
.maxWorker(1)
2525
.build());

0 commit comments

Comments
 (0)