[tokenizers] Fixes ZeroShotClassificationTranslator bug (#3581)

deepjavalibrary · Jan 15, 2025 · a4e61ea · a4e61ea
1 parent 20af02a
commit a4e61ea
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/...src/main/java/ai/djl/modality/nlp/translator/ZeroShotClassificationServingTranslator.java b/...src/main/java/ai/djl/modality/nlp/translator/ZeroShotClassificationServingTranslator.java
@@ -32,9 +32,9 @@ public class ZeroShotClassificationServingTranslator
     private Translator<ZeroShotClassificationInput, ZeroShotClassificationOutput> translator;
 
     /**
-     * Constructs a {@code TokenClassificationServingTranslator} instance.
+     * Constructs a {@code ZeroShotClassificationServingTranslator} instance.
      *
-     * @param translator a {@code Translator} processes token classification input
+     * @param translator a {@code Translator} processes zero-shot-classification input
      */
     public ZeroShotClassificationServingTranslator(
             Translator<ZeroShotClassificationInput, ZeroShotClassificationOutput> translator) {

diff --git a/...enizers/src/main/java/ai/djl/huggingface/translator/ZeroShotClassificationTranslator.java b/...enizers/src/main/java/ai/djl/huggingface/translator/ZeroShotClassificationTranslator.java
@@ -116,7 +116,7 @@ private String applyTemplate(String template, String arg) {
             return template + arg;
         }
         int len = template.length();
-        return template.substring(0, pos) + arg + template.substring(pos + 1, len);
+        return template.substring(0, pos) + arg + template.substring(pos + 2, len);
     }
 
     /**