vickeee465 commited on
Commit
6ba52eb
·
1 Parent(s): c8f14da
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -110,6 +110,10 @@ def predict_wrapper(text, language):
110
  results.append([sentence, label, probability])
111
  results_heatmap.append({"sentence":sentence, "emotions":probs})
112
 
 
 
 
 
113
  figure = plot_emotion_heatmap(prepare_heatmap_data(results_heatmap))
114
  output_info = f'Prediction was made using the <a href="https://huggingface.co/{model_id}">{model_id}</a> model.'
115
  return results, figure, output_info
 
110
  results.append([sentence, label, probability])
111
  results_heatmap.append({"sentence":sentence, "emotions":probs})
112
 
113
+ # let's see...
114
+ print(results)
115
+ print(results_heatmap)
116
+
117
  figure = plot_emotion_heatmap(prepare_heatmap_data(results_heatmap))
118
  output_info = f'Prediction was made using the <a href="https://huggingface.co/{model_id}">{model_id}</a> model.'
119
  return results, figure, output_info