Spaces:
Running
Running
File size: 423 Bytes
4303018 82e2a45 4303018 82e2a45 4303018 82e2a45 4303018 |
1 2 3 4 5 6 7 8 9 10 11 |
from transformers import RobertaTokenizer, RobertaForSequenceClassification
# load tokenizer and model weights
tokenizer = RobertaTokenizer.from_pretrained('SkolkovoInstitute/roberta_toxicity_classifier')
model = RobertaForSequenceClassification.from_pretrained('SkolkovoInstitute/roberta_toxicity_classifier')
# prepare the input
batch = tokenizer.encode('you are amazing', return_tensors='pt')
# inference
model(batch) |