[rust] Adds text classification models to Rust model zoo (#3444)

deepjavalibrary · Aug 30, 2024 · 50e2770 · 50e2770
1 parent d9a45ff
commit 50e2770
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/extensions/tokenizers/src/main/java/ai/djl/engine/rust/zoo/RsModelZoo.java b/extensions/tokenizers/src/main/java/ai/djl/engine/rust/zoo/RsModelZoo.java
@@ -68,6 +68,7 @@ private void init() {
                 if (!initialized) {
                     Version version = new Version(Engine.getDjlVersion());
                     addModels(Application.NLP.TEXT_EMBEDDING, version);
+                    addModels(Application.NLP.TEXT_CLASSIFICATION, version);
                     initialized = true;
                 }
             }