Spaces:
Running
Running
kovacsvi
commited on
Commit
·
18797a3
1
Parent(s):
ba0fb9c
removed filtering logic
Browse files
app.py
CHANGED
@@ -176,17 +176,9 @@ def plot_average_emotion_barplot(heatmap_data):
|
|
176 |
labels = [id2label[i] for i in range(len(mean_scores))]
|
177 |
scores = mean_scores
|
178 |
|
179 |
-
# Optional: filter out near-zero average emotions
|
180 |
-
labels_filtered = []
|
181 |
-
scores_filtered = []
|
182 |
-
for l, s in zip(labels, scores):
|
183 |
-
if s > 0.01:
|
184 |
-
labels_filtered.append(l)
|
185 |
-
scores_filtered.append(s)
|
186 |
-
|
187 |
# Sort for better visualization
|
188 |
sorted_data = sorted(
|
189 |
-
zip(labels_filtered,
|
190 |
)
|
191 |
sorted_labels, sorted_scores = zip(*sorted_data)
|
192 |
|
|
|
176 |
labels = [id2label[i] for i in range(len(mean_scores))]
|
177 |
scores = mean_scores
|
178 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
179 |
# Sort for better visualization
|
180 |
sorted_data = sorted(
|
181 |
+
zip(labels_filtered, scores), key=lambda x: x[1], reverse=True
|
182 |
)
|
183 |
sorted_labels, sorted_scores = zip(*sorted_data)
|
184 |
|