Pant0x commited on
Commit
5024c86
·
verified ·
1 Parent(s): 8b8dfab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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 = Wav2Vec2Processor.from_pretrained(voice_model_name)
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)