From e21a11fc54e32c0211bc8946e5058602121b3d8e Mon Sep 17 00:00:00 2001
From: carlosuc3m <100329787@alumnos.uc3m.es>
Date: Thu, 3 Oct 2024 13:40:46 +0200
Subject: [PATCH] properly close model

---
 .../modelrunner/pytorch/javacpp/PytorchJavaCPPInterface.java    | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/main/java/io/bioimage/modelrunner/pytorch/javacpp/PytorchJavaCPPInterface.java b/src/main/java/io/bioimage/modelrunner/pytorch/javacpp/PytorchJavaCPPInterface.java
index 3597fdb..f0dafca 100644
--- a/src/main/java/io/bioimage/modelrunner/pytorch/javacpp/PytorchJavaCPPInterface.java
+++ b/src/main/java/io/bioimage/modelrunner/pytorch/javacpp/PytorchJavaCPPInterface.java
@@ -367,6 +367,7 @@ else if (task.status == TaskStatus.FAILED)
 			else if (task.status == TaskStatus.CRASHED)
 				throw new RuntimeException();
 			this.runner.close();
+			this.runner = null;
 			return;
 		} else if (this.interprocessing) {
 			return;
@@ -375,6 +376,7 @@ else if (task.status == TaskStatus.CRASHED)
 			return;
 		model.close();
 		model.deallocate();
+		model = null;
 	}
 
 	/**