Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -4,6 +4,7 @@ import random
|
|
| 4 |
import re
|
| 5 |
import torch
|
| 6 |
from transformers import Wav2Vec2ForSequenceClassification, Wav2Vec2Processor
|
|
|
|
| 7 |
import librosa
|
| 8 |
from gtts import gTTS
|
| 9 |
import numpy as np
|
|
@@ -62,7 +63,7 @@ def is_mental_health_related(text: str) -> bool:
|
|
| 62 |
# Load voice emotion model
|
| 63 |
voice_model_name = "Hatman/audio-emotion-detection"
|
| 64 |
voice_model = Wav2Vec2ForSequenceClassification.from_pretrained(voice_model_name)
|
| 65 |
-
voice_processor =
|
| 66 |
|
| 67 |
def detect_voice_emotion(audio_file):
|
| 68 |
audio, sr = librosa.load(audio_file, sr=16000)
|
|
|
|
| 4 |
import re
|
| 5 |
import torch
|
| 6 |
from transformers import Wav2Vec2ForSequenceClassification, Wav2Vec2Processor
|
| 7 |
+
from transformers import Wav2Vec2FeatureExtractor
|
| 8 |
import librosa
|
| 9 |
from gtts import gTTS
|
| 10 |
import numpy as np
|
|
|
|
| 63 |
# Load voice emotion model
|
| 64 |
voice_model_name = "Hatman/audio-emotion-detection"
|
| 65 |
voice_model = Wav2Vec2ForSequenceClassification.from_pretrained(voice_model_name)
|
| 66 |
+
voice_processor = Wav2Vec2FeatureExtractor.from_pretrained(voice_model_name)
|
| 67 |
|
| 68 |
def detect_voice_emotion(audio_file):
|
| 69 |
audio, sr = librosa.load(audio_file, sr=16000)
|