Edit model card

ONNX version of dslim/bert-large-NER

This model is a conversion of dslim/bert-large-NER to ONNX format using the 🤗 Optimum library.

bert-large-NER is a fine-tuned BERT model that is ready to use for Named Entity Recognition and achieves state-of-the-art performance for the NER task. It has been trained to recognize four types of entities: location (LOC), organizations (ORG), person (PER) and Miscellaneous (MISC).

Specifically, this model is a bert-large-cased model that was fine-tuned on the English version of the standard CoNLL-2003 Named Entity Recognition dataset.

Usage

Loading the model requires the 🤗 Optimum library installed.

from optimum.onnxruntime import ORTModelForTokenClassification
from transformers import AutoTokenizer, pipeline


tokenizer = AutoTokenizer.from_pretrained("laiyer/bert-large-NER-onnx")
model = ORTModelForTokenClassification.from_pretrained("laiyer/bert-large-NER-onnx")
ner = pipeline(
    task="ner",
    model=model,
    tokenizer=tokenizer,
)

ner_output = ner("My name is John Doe.")
print(ner_output)

LLM Guard

Anonymize scanner

Community

Join our Slack to give us feedback, connect with the maintainers and fellow users, ask questions, or engage in discussions about LLM security!

Downloads last month
15
Inference Examples
Inference API (serverless) has been turned off for this model.

Model tree for protectai/bert-large-NER-onnx

Quantized
(1)
this model

Dataset used to train protectai/bert-large-NER-onnx

Collection including protectai/bert-large-NER-onnx

Evaluation results