Spaces:

almalabs
/

Article-emotions-categorization

Runtime error

abdulmatinomotoso commited on Feb 22, 2023

Commit

27f0b7c

•

1 Parent(s): dfbd3c2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ labels = labels = ['Comment (Expert / Leadership)', 'Personal News','Event Parti
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-model_name = 'abdulmatinomotoso/finetuned-distilbert-shidhant-emotion-article-categorization'
 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 = 'almalabs/finetuned-distilbert-article-emotions-categorization'
 model = AutoModelForSequenceClassification.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)