z-uo's picture
Update README.md
7ab0e44 verified
metadata
language:
  - it
license: mit
widget:
  - text: 'Ciao, sono Giacomo. Vivo a Milano e lavoro da Armani. '
    example_title: Example 1
  - text: 'Domenica andrò allo stadio con Giovanna a guardare la Fiorentina. '
    example_title: Example 2
base_model: nickprock/bert-italian-finetuned-ner
pipeline_tag: token-classification

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)