Spaces:
Sleeping
Sleeping
vickeee465
commited on
Commit
·
6ba52eb
1
Parent(s):
c8f14da
??? (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
|