huspacy

interfaces/ner.py

@@ -30,7 +30,7 @@ def build_spacy_path(language: str):
 
 def named_entity_recognition(text, language):
     model_id = build_spacy_path(language)
-    pipeline = spacy.load(model_id)
+    pipeline = huspacy.load() if model_id.startswith("hu") else spacy.load(model_id)
     doc = pipeline(text)
     entities = [(ent.text, ent.label_) for ent in doc.ents]
     model_id_hf = f"huspacy/{model_id}" if model_id.startswith("hu") else f"spacy/{model_id}"