fix: model size parameter name

ICIJ · Nov 20, 2024 · cdf6998 · cdf6998
1 parent fe4d205
commit cdf6998
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/datashare-app/src/main/java/org/icij/datashare/nlp/NlpHelper.java b/datashare-app/src/main/java/org/icij/datashare/nlp/NlpHelper.java
@@ -6,7 +6,7 @@
 public class NlpHelper {
     public static Map<String, Object> pipelineExtras(Pipeline.Type pipeline) {
         if (pipeline == Pipeline.Type.SPACY) {
-            return Map.of("size", "md");
+            return Map.of("modelSize", "md");
         }
         return Map.of();
     }