Spaces:

valurank
/

multi-label-emotion

Sleeping

abdulmatinomotoso commited on Sep 1, 2023

Commit

d886294

•

1 Parent(s): 3f27f6d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ import torch
 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 = "valurank/finetuned-distilbert-multi-label-emotion"
 model = AutoModelForSequenceClassification.from_pretrained(model_name)

 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 = "valurank/finetuned-distilbert-multi-label-emotion"
 model = AutoModelForSequenceClassification.from_pretrained(model_name)