correct error sending model source to other process

Author: carlosuc3m

src/main/java/io/bioimage/modelrunner/pytorch/JavaWorker.java

@@ -87,7 +87,7 @@ private void executeScript(String script, Map<String, Object> inputs) {
 		this.reportLaunch();
 		try {
 			if (script.equals("loadModel")) {
-				pi.loadModel((String) inputs.get("modelFolder"), null);
+				pi.loadModel((String) inputs.get("modelFolder"), (String) inputs.get("modelSource"));
 			} else if (script.equals("inference")) {
 				pi.runFromShmas((List<String>) inputs.get("inputs"), (List<String>) inputs.get("outputs"));
 			} else if (script.equals("close")) {

src/main/java/io/bioimage/modelrunner/pytorch/PytorchInterface.java

@@ -100,6 +100,8 @@ public class PytorchInterface implements DeepLearningEngineInterface {
 	 */
 	private ZooModel<NDList, NDList> model;
 
+	private String modelFolder;
+	
 	private String modelSource;
 
 	private boolean interprocessing = true;
@@ -155,6 +157,7 @@ private Service getRunner() throws IOException, URISyntaxException {
 	public void loadModel(String modelFolder, String modelSource)
 		throws LoadModelException
 	{
+		this.modelFolder = modelFolder;
 		this.modelSource = modelSource;
 		if (interprocessing) {
 			try {
@@ -188,7 +191,8 @@ public void loadModel(String modelFolder, String modelSource)
 
 	private void launchModelLoadOnProcess() throws IOException, InterruptedException {
 		HashMap<String, Object> args = new HashMap<String, Object>();
-		args.put("modelFolder", modelSource);
+		args.put("modelFolder", modelFolder);
+		args.put("modelSource", modelSource);
 		Task task = runner.task("loadModel", args);
 		task.waitFor();
 		if (task.status == TaskStatus.CANCELED)