Update app.py

app.py

@@ -10,9 +10,9 @@ from datasets import load_dataset
 
 
 def load_model():
-    model = AutoModelForSequenceClassification.from_pretrained('model_distilbert_trained', use_auth_token=True)
+    model = AutoModelForSequenceClassification.from_pretrained('model_roberta_trained', use_auth_token=True)
     tokenizer = AutoTokenizer.from_pretrained(
-        'distilbert-base-cased', do_lower_case=True)
+        'roberta-base', do_lower_case=True)
     model.eval()
     return model, tokenizer