Spaces:

anitalp
/

NLP_Models_sequence

Sleeping

NLP_Models_sequence / app.py

Update app.py

4303018 over 2 years ago

No virus

423 Bytes

	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)