Replacing the base model

[golankai]

Hello!
Is it possible to replace the base NER model that the tool uses in the identifying phase?
I want to try it with a new model trained on my data.
Thank you very much!

[talbaumel]

Hi, you can find the model in this repo
Hope it helps :)

[omri374]

Adding on @talbaumel's message, as Hebsafeharbor is built on top of Microsoft Presidio, you can plug in any NER model through Presidio. Specifically, there is native support for spacy, huggingface and stanza models.
This is where the model is loaded:

HebSafeHarbor/hebsafeharbor/identifier/phi_identifier.py

Line 80 in 4a0ba72

nlp_engine = HebSpacyNlpEngine(models={"he": "he_ner_news_trf"})

More info on customizing the NER model in Presidio can be found here: https://microsoft.github.io/presidio/analyzer/customizing_nlp_models/