Views
No views yet
1from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassification
2
3tokenizer = AutoTokenizer.from_pretrained("erst/xlm-roberta-base-finetuned-nace")
4model = AutoModelForSequenceClassification.from_pretrained("erst/xlm-roberta-base-finetuned-nace")
5
6pl = pipeline(
7 "sentiment-analysis",
8 model=model,
9 tokenizer=tokenizer,
10 return_all_scores=False,
11)
12
13pl("The purpose of our company is to build houses")