Home

Contents

• API

API

This is a simple example of API usage:

from bulk_ner.api import NERAnnotator
from bulk_ner.src.service_dynamic import dynamic_init


ner_model = dynamic_init(src_dir=join(TestTransformersNERPipeline.CURRENT_DIR, "../models"),
                                 class_filepath="dp_130.py",
                                 class_name="DeepPavlovNER")(model="ner_ontonotes_bert")

annotator = NERAnnotator(ner_model=ner_model, chunk_limit=128)

data_it = annotator.iter_annotated_data(
    data_dict_it=[{"text": [TestTransformersNERPipeline.text]}],
    prompt="{text}")
    
for data in data_it:
    # Handle your NER data here ...