Hugging Face's logo

Spaces:
Sephfox
/
Cain
Sleeping

App Files Community
1
Sephfox commited on
Commit
1978fc5
·
verified ·
1 Parent(s): aaf362d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ emotions_target = pd.Categorical(df['emotion']).codes
43
  emotion_classes = pd.Categorical(df['emotion']).categories
44
 
45
  # Load pre-trained BERT model for emotion prediction
46
- emotion_prediction_model = AutoModelForSequenceClassification.from_pretrained("bhadresh-savani/distilbert-base-go-emotions-emotion")
47
 
48
  # Lazy loading for the fine-tuned language model
49
  _finetuned_lm_tokenizer = None
 
43
  emotion_classes = pd.Categorical(df['emotion']).categories
44
 
45
  # Load pre-trained BERT model for emotion prediction
46
+ emotion_prediction_model = AutoModelForSequenceClassification.from_pretrained("bhadresh-savani/distilbert-base-uncased-emotion")
47
 
48
  # Lazy loading for the fine-tuned language model
49
  _finetuned_lm_tokenizer = None