Merge remote-tracking branch 'origin/main'

holunda-io · Apr 18, 2024 · a06d8b5 · a06d8b5
2 parents fc5d26c + ee5eeab
commit a06d8b5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bpm_ai_inference/token_classification/gliner_token_classifier.py b/bpm_ai_inference/token_classification/gliner_token_classifier.py
@@ -20,7 +20,7 @@ class GlinerTokenClassifier(ZeroShotTokenClassifier):
     To use, you should have the ``gliner`` python package installed.
     """
 
-    def __init__(self, model: str = "urchade/gliner_large"):
+    def __init__(self, model: str = "urchade/gliner_large-v2.1"):
         self.model = GLiNER.from_pretrained(model)
 
     @override
@@ -47,4 +47,4 @@ async def _do_classify(
                 )
                 for e in entities
             ]
-        )
+        )