Spaces:

valurank
/

multi-label-emotion

Sleeping

abdulmatinomotoso commited on Apr 7, 2023

Commit

b578d51

1 Parent(s): 48e42af

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ target_list = ["Cute", "Infuriating", "Sentimental", "Empathetic",
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-model_name = 'abdulmatinomotoso/finetuned-distilbert-multi-label-emotion_5'
 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 = 'abdulmatinomotoso/finetuned-distilbert-multi-label-emotion_6'
 model = AutoModelForSequenceClassification.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)