diff --git a/bpm_ai_inference/token_classification/gliner_token_classifier.py b/bpm_ai_inference/token_classification/gliner_token_classifier.py index 1528f7a..9835a21 100644 --- 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 ] - ) \ No newline at end of file + )