Cristian commited on
Commit
16226cc
1 Parent(s): 1b3e839
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ with gr.Blocks() as demo:
41
  label_value = f"{l['label']} [{l['score']}]"
42
  label_value_t5 = get_emotion(bot_message)
43
 
44
- s = sentiment_task(bot_message)
45
  sentiment_value = f"{s['label']} [{s['score']}]"
46
 
47
  return "", chat_history, f"Emotion [1]: {label_value_t5} - Emotion [2]: {label_value} - Sentiment : {sentiment_value}"
 
41
  label_value = f"{l['label']} [{l['score']}]"
42
  label_value_t5 = get_emotion(bot_message)
43
 
44
+ s = sentiment_task(bot_message)[0]
45
  sentiment_value = f"{s['label']} [{s['score']}]"
46
 
47
  return "", chat_history, f"Emotion [1]: {label_value_t5} - Emotion [2]: {label_value} - Sentiment : {sentiment_value}"