This is an automated email from the ASF dual-hosted git repository.

tallison pushed a commit to branch branch_2x
in repository https://gitbox.apache.org/repos/asf/tika.git


The following commit(s) were added to refs/heads/branch_2x by this push:
     new 2384816ec TIKA-4239 -- band aid clear the cache for jenkins
2384816ec is described below

commit 2384816ecac17d8157b240a5f4a62dfe3d052fa5
Author: tallison <[email protected]>
AuthorDate: Mon Dec 2 08:30:15 2024 -0500

    TIKA-4239 -- band aid clear the cache for jenkins
---
 .../src/main/java/org/apache/tika/dl/imagerec/DL4JVGG16Net.java      | 5 +++++
 1 file changed, 5 insertions(+)

diff --git 
a/tika-parsers/tika-parsers-ml/tika-dl/src/main/java/org/apache/tika/dl/imagerec/DL4JVGG16Net.java
 
b/tika-parsers/tika-parsers-ml/tika-dl/src/main/java/org/apache/tika/dl/imagerec/DL4JVGG16Net.java
index a6f6ca81a..327dbb17e 100644
--- 
a/tika-parsers/tika-parsers-ml/tika-dl/src/main/java/org/apache/tika/dl/imagerec/DL4JVGG16Net.java
+++ 
b/tika-parsers/tika-parsers-ml/tika-dl/src/main/java/org/apache/tika/dl/imagerec/DL4JVGG16Net.java
@@ -26,6 +26,7 @@ import java.util.List;
 import java.util.Map;
 import java.util.Set;
 
+import org.apache.commons.io.FileUtils;
 import org.datavec.image.loader.NativeImageLoader;
 import org.deeplearning4j.nn.graph.ComputationGraph;
 import org.deeplearning4j.util.ModelSerializer;
@@ -98,6 +99,10 @@ public class DL4JVGG16Net implements ObjectRecogniser {
         try {
             if (serialize) {
                 if (cacheDir.exists()) {
+                    FileUtils.deleteDirectory(cacheDir);
+                }
+                if (cacheDir.exists()) {
+                    LOG.info("Trying to load preprocessed model from {}", 
cacheDir);
                     model = ModelSerializer.restoreComputationGraph(cacheDir);
                     LOG.info("Preprocessed Model Loaded from {}", cacheDir);
                 } else {

Reply via email to