Views
No views yet
1from transformers import pipeline
2
3
4model_path = 'marianna13/xlm-roberta-fine-tuned-on-russian-abusive-language'
5
6id2label = {
7 0:'неопасный тескт',
8 1:'опасный тескт'
9}
10
11label2id = {
12 'неопасный тескт':0,
13 'опасный тескт':1
14}
15config = AutoConfig.from_pretrained(model_path, id2label=id2label, label2id=label2id)
16tokenizer = AutoTokenizer.from_pretrained(model_path)
17
18model = AutoModelForSequenceClassification.from_pretrained(model_path, config=config)
19
20text = "Прекрасный день."
21pipe = pipeline('text-classification', model=model, tokenizer=tokenizer)
22pipe(text)[{'label': 'неопасный текcт', 'score': 0.9249424934387207}]