kovacsvi commited on
Commit
18797a3
·
1 Parent(s): ba0fb9c

removed filtering logic

Browse files
Files changed (1) hide show
  1. app.py +1 -9
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, scores_filtered), key=lambda x: x[1], reverse=True
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