cspocketindia commited on
Commit
ce6c474
1 Parent(s): a6ff391

update hf model

Browse files
Files changed (1) hide show
  1. gradio_app.py +1 -1
gradio_app.py CHANGED
@@ -3,7 +3,7 @@ import gradio
3
  from src.models.bert import BERTClassifier
4
  from src.utils.utilities import Utility
5
 
6
- model = BERTClassifier(model_name='Bert_GoEmotions_4Epochs.model')
7
 
8
  classes = Utility().read_emotion_list()
9
 
 
3
  from src.models.bert import BERTClassifier
4
  from src.utils.utilities import Utility
5
 
6
+ model = BERTClassifier(model_name='jeevavijay10/nlp-goemotions-bert')
7
 
8
  classes = Utility().read_emotion_list()
9