Spaces:

mskov
/

Speech-Trigger-Detection

Runtime error

mskov commited on Sep 21, 2023

Commit

b5b947d

1 Parent(s): 6d9a5de

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -110,7 +110,7 @@ def classify_toxicity(audio_file, classify_anxiety, emo_class, explitive_selecti
     # 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_dict = {label: score for label, score in zip(classification_output['labels'], classification_output['scores'])}
-    k = max(label_score_dict, value=label_score_dict.get)
     maxval = label_score_dict[k]
     if maxval > tox_score:
         if maxval > threshold:

     # 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_dict = {label: score for label, score in zip(classification_output['labels'], classification_output['scores'])}
+    k = max(label_score_dict, key=label_score_dict.get)
     maxval = label_score_dict[k]
     if maxval > tox_score:
         if maxval > threshold: