Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -40,10 +40,10 @@ class_options = {
|
|
| 40 |
|
| 41 |
pipe = pipeline("automatic-speech-recognition", model="openai/whisper-large")
|
| 42 |
|
| 43 |
-
def classify_emotion():
|
| 44 |
#### Emotion classification ####
|
| 45 |
emotion_classifier = foreign_class(source="speechbrain/emotion-recognition-wav2vec2-IEMOCAP", pymodule_file="custom_interface.py", classname="CustomEncoderWav2vec2Classifier")
|
| 46 |
-
out_prob, score, index, text_lab = emotion_classifier.classify_file(
|
| 47 |
emostr = "emo class"
|
| 48 |
return emo_dict[text_lab[0]], emostr
|
| 49 |
|
|
@@ -67,7 +67,7 @@ def classify_toxicity(audio_file, text_input, classify_anxiety, emo_class):
|
|
| 67 |
print(toxicity_score)
|
| 68 |
# emo call
|
| 69 |
if emo_class != None:
|
| 70 |
-
classify_emotion()
|
| 71 |
#### Text classification #####
|
| 72 |
|
| 73 |
device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
|
|
|
|
| 40 |
|
| 41 |
pipe = pipeline("automatic-speech-recognition", model="openai/whisper-large")
|
| 42 |
|
| 43 |
+
def classify_emotion(audio):
|
| 44 |
#### Emotion classification ####
|
| 45 |
emotion_classifier = foreign_class(source="speechbrain/emotion-recognition-wav2vec2-IEMOCAP", pymodule_file="custom_interface.py", classname="CustomEncoderWav2vec2Classifier")
|
| 46 |
+
out_prob, score, index, text_lab = emotion_classifier.classify_file(audio)
|
| 47 |
emostr = "emo class"
|
| 48 |
return emo_dict[text_lab[0]], emostr
|
| 49 |
|
|
|
|
| 67 |
print(toxicity_score)
|
| 68 |
# emo call
|
| 69 |
if emo_class != None:
|
| 70 |
+
classify_emotion(audio_file)
|
| 71 |
#### Text classification #####
|
| 72 |
|
| 73 |
device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
|