cspocketindia commited on
Commit
760ecf9
1 Parent(s): ce6c474

Update field name

Browse files
Files changed (1) hide show
  1. gradio_app.py +3 -3
gradio_app.py CHANGED
@@ -7,9 +7,9 @@ model = BERTClassifier(model_name='jeevavijay10/nlp-goemotions-bert')
7
 
8
  classes = Utility().read_emotion_list()
9
 
10
- def predict(inputSentence):
11
- print(inputSentence)
12
- predictions = model.evaluate([inputSentence])
13
  print(f"Predictions: {predictions}")
14
  return classes[predictions[0]]
15
 
 
7
 
8
  classes = Utility().read_emotion_list()
9
 
10
+ def predict(sentence):
11
+ print(sentence)
12
+ predictions = model.evaluate([sentence])
13
  print(f"Predictions: {predictions}")
14
  return classes[predictions[0]]
15