Update app.py

app.py

@@ -1,11 +1,7 @@
-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')
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+  
+tokenizer = AutoTokenizer.from_pretrained("SkolkovoInstitute/roberta_toxicity_classifier")
 
-# inference
-model(batch)
+model = AutoModelForSequenceClassification.from_pretrained("SkolkovoInstitute/roberta_toxicity_classifier")