From 50e2770012b325a70d1bc0c2547f760d9d2f50f6 Mon Sep 17 00:00:00 2001
From: Frank Liu <frankfliu2000@gmail.com>
Date: Fri, 30 Aug 2024 12:33:01 -0700
Subject: [PATCH] [rust] Adds text classification models to Rust model zoo
 (#3444)

---
 .../src/main/java/ai/djl/engine/rust/zoo/RsModelZoo.java         | 1 +
 1 file changed, 1 insertion(+)

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
index f706bc86542..6bd7cfc9d17 100644
--- 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;
                 }
             }