Update app.py

app.py

@@ -87,11 +87,23 @@ def classify_toxicity(audio_file, selected_sounds, slider):
     scores = average_logprobs.softmax(-1).tolist()
     
     class_score_dict = {class_name: score for class_name, score in zip(class_names, scores)}
-    
+    matching_label_score = {}
+    exceeding_threshold = []
     for selected_class_name in selected_sounds:
         print(selected_sounds)
         if selected_class_name in class_score_dict:
-            score = class_score_dict[selected_class_name]
+            #score = class_score_dict[selected_class_name]
+            matching_label_score[class_name] = label_score_dict[class_name]
+            for label, score in matching_label_score.items():
+                # Check if the score is greater than the threshold
+                if score > threshold:
+                    # If it exceeds the threshold, append the label and score to the list
+                    exceeding_threshold.append((label, score))
+                    affirm = f"Threshold exceeded with "{exceeding_threshold}
+                else:
+                    affirm = ""
+                
+'''
             print(score)
             if score > threshold:
                 print(f"Threshold exceeded for class '{selected_class_name}': Score = {score:.4f}")
@@ -99,6 +111,7 @@ def classify_toxicity(audio_file, selected_sounds, slider):
             else:
                 print("under threshold")
                 affirm = ""
+'''
     miso_label_data= {class_name: score for class_name, score in zip(class_names, scores)}
     # miso_label_dict = {label: score for label, score in classify_anxiety[0].items()}