Update app.py

app.py

@@ -112,7 +112,9 @@ def classify_toxicity(audio_file, text_input, classify_anxiety, emo_class, expli
         print("keys ", classification_output.keys())
         
         # formatted_classification_output = "\n".join([f"{key}: {value}" for key, value in classification_output.items()])
-        label_score_pairs = [(label, score) for label, score in zip(classification_output['labels'], classification_output['scores'])]
+        # label_score_pairs = [(label, score) for label, score in zip(classification_output['labels'], classification_output['scores'])]
+        label_score_dict = {label: score for label, score in zip(classification_output['labels'], classification_output['scores'])}
+
 
 
 
@@ -124,7 +126,7 @@ def classify_toxicity(audio_file, text_input, classify_anxiety, emo_class, expli
         else:
             affirm = ""
  
-        return toxicity_score, label_score_pairs, transcribed_text, affirm
+        return toxicity_score, label_score_dict, transcribed_text, affirm
         # return f"Toxicity Score ({available_models[selected_model]}): {toxicity_score:.4f}"
     else: 
         threshold = slider_logic(slider)