SHIROI-07 commited on
Commit
b715cee
·
verified ·
1 Parent(s): b50f40d

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -3,7 +3,7 @@ import torch
3
 
4
  class EmotionModel:
5
  def __init__(self):
6
- self.model_name = "distilbert-base-uncased-finetuned-sst-2-english model"
7
  self.tokenizer = AutoTokenizer.from_pretrained(self.model_name)
8
  self.model = AutoModelForSequenceClassification.from_pretrained(self.model_name)
9
  self.labels = self.model.config.id2label
 
3
 
4
  class EmotionModel:
5
  def __init__(self):
6
+ self.model_name = "joeddav/distilbert-base-uncased-go-emotions-student"
7
  self.tokenizer = AutoTokenizer.from_pretrained(self.model_name)
8
  self.model = AutoModelForSequenceClassification.from_pretrained(self.model_name)
9
  self.labels = self.model.config.id2label