diff --git a/src/main/java/io/bioimage/modelrunner/pytorch/javacpp/JavaWorker.java b/src/main/java/io/bioimage/modelrunner/pytorch/javacpp/JavaWorker.java index 6e5d7e5..b46f7c7 100644 --- a/src/main/java/io/bioimage/modelrunner/pytorch/javacpp/JavaWorker.java +++ b/src/main/java/io/bioimage/modelrunner/pytorch/javacpp/JavaWorker.java @@ -87,10 +87,10 @@ private void executeScript(String script, Map inputs) { this.reportLaunch(); try { if (script.equals("loadModel")) { - System.out.println("STATY IN WORKER LOAD"); + update("STATY IN WORKER LOAD", null, null); pi.loadModel((String) inputs.get("modelFolder"), (String) inputs.get("modelSource")); } else if (script.equals("inference")) { - System.out.println("STATY IN WORKER"); + update("STATY IN WORKER", null, null); pi.runFromShmas((List) inputs.get("inputs"), (List) inputs.get("outputs")); } else if (script.equals("close")) { pi.closeModel();