Views
No views yet
pip install optimum onnxruntime onnx1import time
2from transformers import AutoTokenizer, pipeline
3from optimum.onnxruntime import ORTModelForTokenClassification
4
5tokenizer = AutoTokenizer.from_pretrained("z-uo/bert-italian-ner-onnx-quantized-avx512")
6model = ORTModelForTokenClassification.from_pretrained("z-uo/bert-italian-ner-onnx-quantized-avx512")
7nerpipeline = pipeline('ner', model=model, tokenizer=tokenizer)
8
9text = "La sede storica della Olivetti è ad Ivrea"
10output = nerpipeline(text)