Hugging Face's logo

Spaces:
sk23aib
/
Text_Emotion_Multi_model
Sleeping

App Files Community
sk23aib commited on
Commit
ba3bb7f
·
verified ·
1 Parent(s): a94fd8b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -9,9 +9,9 @@ from transformers import (
9
 
10
  # Model repositories on Hugging Face Hub
11
  model_repos = {
12
- "BERT": "sk23aib/emotion-bert",
13
- "XLNet": "sk23aib/emotion-xlnet",
14
- "GPT-2": "sk23aib/emotion-gpt2"
15
  }
16
 
17
  # Emotion labels (must match model training order)
 
9
 
10
  # Model repositories on Hugging Face Hub
11
  model_repos = {
12
+ "BERT": "sk23aib/Bert_emotion_model",
13
+ "XLNet": "sk23aib/Xlnet_emotion_model",
14
+ "GPT-2": "sk23aib/Gpt2_emotion_model"
15
  }
16
 
17
  # Emotion labels (must match model training order)