Views
No views yet
1from transformers import pipeline, AutoTokenizer
2from optimum.onnxruntime import ORTModelForSequenceClassification
3
4model_id = "dilexsan/bertweet_base_sentimental_onnx"
5
6tokenizer = AutoTokenizer.from_pretrained(model_id)
7
8model = ORTModelForSequenceClassification.from_pretrained(model_id)
9
10sentiment_analyzer = pipeline(
11 "text-classification",
12 model=model,
13 tokenizer=tokenizer
14)
15
16text = "I am so happy with this service, it was great!"
17result = sentiment_analyzer(text)
18
19print(result)