fix(plugin): remove tensorflow engine

BakaBotTeam · Oct 3, 2024 · c6dd5cb · c6dd5cb
1 parent 53e6758
commit c6dd5cb
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 5 deletions.
diff --git a/build.gradle.kts b/build.gradle.kts
@@ -27,11 +27,6 @@ dependencies {
     implementation("org.apache.commons:commons-lang3:3.16.0")
     implementation("org.apache.commons:commons-imaging:1.0-alpha1")
     implementation("ai.djl.onnxruntime:onnxruntime-engine:0.23.0")
-    implementation(platform("ai.djl:bom:0.23.0"))
-    implementation("ai.djl.tensorflow:tensorflow-engine:0.23.0")
-    implementation("ai.djl.tensorflow:tensorflow-model-zoo:0.23.0")
-    implementation("ai.djl.tensorflow:tensorflow-native-cpu:2.10.0:win-x86_64")
-    implementation("ai.djl.tensorflow:tensorflow-native-cpu:2.10.0:linux-x86_64")
 
     compileOnly("top.mrxiaom.mirai:overflow-core-api:$overflow_version")
     compileOnly("top.mrxiaom.mirai:overflow-core:$overflow_version")

diff --git a/src/main/java/ltd/guimc/dlm/DLModel.java b/src/main/java/ltd/guimc/dlm/DLModel.java
@@ -2,6 +2,7 @@
 
 import ai.djl.MalformedModelException;
 import ai.djl.Model;
+import ai.djl.engine.Engine;
 import ai.djl.inference.Predictor;
 import ai.djl.modality.Classifications;
 import ai.djl.modality.cv.Image;
@@ -22,6 +23,7 @@ public class DLModel {
     private static boolean inited = false;
 
     public static void init() {
+        System.out.println("Available engines: " + Engine.getAllEngines());
         try {
             // 从资源文件中获取模型
             InputStream modelStream = DLModel.class.getResourceAsStream("/Nailong.onnx");

diff --git a/src/test/java/Test.java b/src/test/java/Test.java
@@ -1,5 +1,7 @@
+import ltd.guimc.dlm.DLModel;
 
 public class Test {
     public static void main(String[] args) throws InterruptedException {
+        DLModel.init();
     }
 }