patched custom model deleted bug

jiyuuchc · Jan 20, 2024 · 79539d8 · 79539d8
1 parent 5c60abb
commit 79539d8
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/main/java/fiji/plugin/trackmate/detector/lacss/LacssDetectorFactory.java b/src/main/java/fiji/plugin/trackmate/detector/lacss/LacssDetectorFactory.java
@@ -165,9 +165,12 @@ private void addOnShutdownHook()
 			public void run()
 			{
 				// new File(pyFilePath).delete();
-				new File(modelPath).delete();
-
-				if (pyServer.isAlive()) {
+				if (settings.get(KEY_LACSS_MODEL) == PretrainedModel.Default)
+				{
+					new File(modelPath).delete();
+				}
+				if (pyServer.isAlive()) 
+				{
 					pyServer.destroy();
 				}
 			}