[api] Fix list parsing in CrossEncoderServingTranslator (deepjavalibr…

…ary#3374)
ydm-amazon · Aug 1, 2024 · 22da01c · 22da01c
1 parent 9f5f8ed
commit 22da01c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/src/main/java/ai/djl/modality/nlp/translator/CrossEncoderServingTranslator.java b/api/src/main/java/ai/djl/modality/nlp/translator/CrossEncoderServingTranslator.java
@@ -35,7 +35,7 @@
 /** A {@link Translator} that can handle generic cross encoder {@link Input} and {@link Output}. */
 public class CrossEncoderServingTranslator implements NoBatchifyTranslator<Input, Output> {
 
-    private static final Type LIST_TYPE = new TypeToken<List<String>>() {}.getType();
+    private static final Type LIST_TYPE = new TypeToken<List<StringPair>>() {}.getType();
 
     private Translator<StringPair, float[]> translator;