Spaces:

mskov
/

Speech-Trigger-Detection

Runtime error

mskov commited on Sep 12, 2023

Commit

f8fa917

•

1 Parent(s): 9d36990

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -76,7 +76,7 @@ def classify_toxicity(audio_file, text_input, classify_anxiety):
     #### Emotion classification ####
     emotion_classifier = foreign_class(source="speechbrain/emotion-recognition-wav2vec2-IEMOCAP", pymodule_file="custom_interface.py", classname="CustomEncoderWav2vec2Classifier")
-    out_prob, score, index, text_lab = emotion_classifier.classify_file(audio_file.name)
     return toxicity_score, classification_output, emo_dict[text_lab[0]], transcribed_text
     # return f"Toxicity Score ({available_models[selected_model]}): {toxicity_score:.4f}"

     #### Emotion classification ####
     emotion_classifier = foreign_class(source="speechbrain/emotion-recognition-wav2vec2-IEMOCAP", pymodule_file="custom_interface.py", classname="CustomEncoderWav2vec2Classifier")
+    out_prob, score, index, text_lab = emotion_classifier.classify_file(audio_file)
     return toxicity_score, classification_output, emo_dict[text_lab[0]], transcribed_text
     # return f"Toxicity Score ({available_models[selected_model]}): {toxicity_score:.4f}"