Update app.py

app.py

@@ -2,8 +2,8 @@ from transformers import AutoModelForSequenceClassification, AutoTokenizer, Trai
 import torch
 from datasets import load_dataset
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model = AutoModelForSequenceClassification.from_pretrained("facebook/bart-large-cnn").to(device)
-tokenizer = AutoTokenizer.from_pretrained("facebook/bart-large-cnn")
+model = AutoModelForSequenceClassification.from_pretrained("gpt2").to(device)
+tokenizer = AutoTokenizer.from_pretrained("gpt2")