Update app.py

app.py

@@ -5,13 +5,12 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
 
 
-target_list = ["Cute", "Infuriating", "Sentimental", "Empathetic",
-               "Cynical", "Depressing", "Awe-inspiring", "Patriotic", "Educational",
-               "Encouraging", "Voyeuristic", "Funny", "Sarcastic", "Dismissive", "Disparaging"]
+target_list = ["Playful", "Infuriating", "Sentimental", "Cynical", "Depressing", "Awe-inspiring", "Patriotic", "Begrudging", "Educational", "Hopeful",
+"Sarcastic", "Disrespectful", "Disparaging"]
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 
-model_name = 'abdulmatinomotoso/finetuned-distilbert-multi-label-emotion_6'
+model_name = 'abdulmatinomotoso/finetuned-distilbert-multi-label-emotion_headline'
 model = AutoModelForSequenceClassification.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)