z-uo/bert-italian-ner-onnx-quantized-avx512

Bert Italian NER ONNX avx512

This model is the onnx version of nickprock/bert-italian-finetuned-ner.

To use you need to intall following libraries:

pip install optimum onnxruntime onnx

And run with the following script:

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

tokenizer = AutoTokenizer.from_pretrained("z-uo/bert-italian-ner-onnx-quantized-avx512")
model = ORTModelForTokenClassification.from_pretrained("z-uo/bert-italian-ner-onnx-quantized-avx512")
nerpipeline = pipeline('ner', model=model, tokenizer=tokenizer)

text = "La sede storica della Olivetti è ad Ivrea"
output = nerpipeline(text)