Spaces:
Sleeping
Sleeping
vickeee465
commited on
Commit
·
0d8846e
1
Parent(s):
3fc45cd
??? (2)
Browse files
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
|