mskov commited on
Commit
ed2f0b8
1 Parent(s): 187f615

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -42,10 +42,10 @@ class_options = {
42
  pipe = pipeline("automatic-speech-recognition", model="openai/whisper-large")
43
 
44
  def classify_emotion(audio):
45
- #### Emotion classification ####
46
- emotion_classifier = foreign_class(source="speechbrain/emotion-recognition-wav2vec2-IEMOCAP", pymodule_file="custom_interface.py", classname="CustomEncoderWav2vec2Classifier")
47
- out_prob, score, index, text_lab = emotion_classifier.classify_file(audio)
48
- return emo_dict[text_lab[0]]
49
 
50
  # Create a Gradio interface with audio file and text inputs
51
  def classify_toxicity(audio_file, text_input, classify_anxiety, emo_class, explitive_selection):
@@ -55,7 +55,7 @@ def classify_toxicity(audio_file, text_input, classify_anxiety, emo_class, expli
55
  else:
56
  transcribed_text = text_input
57
  if classify_anxiety != "misophonia":
58
- print("emo_class ", emo_class, "explitive select", explitive_selection)
59
  # explitive call
60
  if replace_explitives != None and emo_class == None:
61
  transcribed_text = replace_explitives.sub_explitives(transcribed_text, explitive_selection)
 
42
  pipe = pipeline("automatic-speech-recognition", model="openai/whisper-large")
43
 
44
  def classify_emotion(audio):
45
+ #### Emotion classification ####
46
+ emotion_classifier = foreign_class(source="speechbrain/emotion-recognition-wav2vec2-IEMOCAP", pymodule_file="custom_interface.py", classname="CustomEncoderWav2vec2Classifier")
47
+ out_prob, score, index, text_lab = emotion_classifier.classify_file(audio)
48
+ return emo_dict[text_lab[0]]
49
 
50
  # Create a Gradio interface with audio file and text inputs
51
  def classify_toxicity(audio_file, text_input, classify_anxiety, emo_class, explitive_selection):
 
55
  else:
56
  transcribed_text = text_input
57
  if classify_anxiety != "misophonia":
58
+ print("emo_class ", emo_class, "explitive select", explitive_selection)
59
  # explitive call
60
  if replace_explitives != None and emo_class == None:
61
  transcribed_text = replace_explitives.sub_explitives(transcribed_text, explitive_selection)