Spaces:

valurank
/

multi-label-emotion

Sleeping

abdulmatinomotoso commited on Apr 28, 2023

Commit

618a85c

1 Parent(s): 3141888

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ target_list = ["Playful", "Infuriating", "Sentimental", "Cynical", "Depressing",
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-model_name = 'abdulmatinomotoso/finetuned-distilbert-multi-label-emotion_headline'
 model = AutoModelForSequenceClassification.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)

 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+model_name = 'valurank/finetuned-distilbert-multi-label-emotion'
 model = AutoModelForSequenceClassification.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)