SamanthaStorm commited on
Commit
c519b97
·
verified ·
1 Parent(s): 9623d13

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -171,7 +171,10 @@ def analyze_composite(msg1, msg2, msg3, *answers_and_none):
171
  results = [analyze_single_message(m, THRESHOLDS.copy(), []) for m in active]
172
  abuse_scores = [r[0] for r in results]
173
  darvo_scores = [r[3] for r in results]
174
- top_label = max({label for r in results for label in r[2]}, key=lambda l: abuse_scores[0])
 
 
 
175
  composite_abuse = int(round(sum(abuse_scores) / len(abuse_scores)))
176
  avg_darvo = round(sum(darvo_scores) / len(darvo_scores), 3)
177
 
 
171
  results = [analyze_single_message(m, THRESHOLDS.copy(), []) for m in active]
172
  abuse_scores = [r[0] for r in results]
173
  darvo_scores = [r[3] for r in results]
174
+ top_label = max(
175
+ [(label, score) for r in results for label, score in r[2]],
176
+ key=lambda x: x[1]
177
+ )[0]
178
  composite_abuse = int(round(sum(abuse_scores) / len(abuse_scores)))
179
  avg_darvo = round(sum(darvo_scores) / len(darvo_scores), 3)
180