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)
- Downloads last month
- 5