[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;
  }
  }