marianna13/xlm-roberta-fine-tuned-on-russian-abusive-language

from transformers import pipeline


model_path = 'marianna13/xlm-roberta-fine-tuned-on-russian-abusive-language'

id2label = {
    0:'неопасный тескт',
    1:'опасный тескт'
}

label2id = {
    'неопасный тескт':0,
    'опасный тескт':1
}
config = AutoConfig.from_pretrained(model_path, id2label=id2label, label2id=label2id)
tokenizer = AutoTokenizer.from_pretrained(model_path)

model = AutoModelForSequenceClassification.from_pretrained(model_path, config=config)

text = "Прекрасный день."
pipe = pipeline('text-classification', model=model, tokenizer=tokenizer)
pipe(text)

[{'label': 'неопасный текcт', 'score': 0.9249424934387207}]