Update app.py

app.py

@@ -96,14 +96,14 @@ def classify_toxicity(audio_file, selected_sounds, slider):
             score = class_score_dict[selected_class_name]
             matching_label_score[selected_class_name] = score
             print("matching label score type is ", type(matching_label_score))
-    
-    #  iterate through matching_label_score to check if any scores exceed the threshold
-    for label, score in matching_label_score.items():
-        if score > threshold:
-            exceeding_threshold.append((label, score))
-            affirm = "Threshold Exceeded"
-        else:
-            affirm = ""
+            
+    highest_score = max(matching_label_scores, key=matching_label_scores.get)
+
+    if highest_score is not None and highest_score > threshold:
+        exceeding_threshold.append((label, score))
+        affirm = "Threshold Exceeded"
+    else:
+        affirm = " "
             
     # miso_label_dict = {label: score for label, score in classify_anxiety[0].items()}