Spaces:

valurank
/

multi-label-emotion

Sleeping

abdulmatinomotoso commited on May 31, 2023

Commit

fee5906

•

1 Parent(s): c836955

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,6 @@ target_list = ["Playful", "Infuriating", "Sentimental", "Cynical", "Depressing",
 "Sarcastic", "Disrespectful", "Disparaging"]
 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)

 "Sarcastic", "Disrespectful", "Disparaging"]
 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)