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 b46f7c7..b696fb0 100644 --- a/src/main/java/io/bioimage/modelrunner/pytorch/javacpp/JavaWorker.java +++ b/src/main/java/io/bioimage/modelrunner/pytorch/javacpp/JavaWorker.java @@ -35,6 +35,7 @@ public static void main(String[] args) { try { pi = new PytorchJavaCPPInterface(false); } catch (IOException | URISyntaxException e) { + e.printStackTrace(); return; } @@ -87,13 +88,16 @@ private void executeScript(String script, Map inputs) { this.reportLaunch(); try { if (script.equals("loadModel")) { - update("STATY IN WORKER LOAD", null, null); + update("STATY IN WORKER LOAD LOAD", null, null); pi.loadModel((String) inputs.get("modelFolder"), (String) inputs.get("modelSource")); } else if (script.equals("inference")) { - update("STATY IN WORKER", null, null); + update("STATY IN WORKER ------------RUN", null, null); pi.runFromShmas((List) inputs.get("inputs"), (List) inputs.get("outputs")); } else if (script.equals("close")) { pi.closeModel(); + } else { + update("LOL WTF", null, null); + update("LOL WTF -------------- " + script, null, null); } } catch(Exception | Error ex) { this.fail(Types.stackTrace(ex));