Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ import torchaudio
|
|
7 |
emotion_labels = ["angry", "calm", "disgust", "fearful", "happy", "neutral", "sad", "surprised"]
|
8 |
|
9 |
# Load model and processor
|
10 |
-
model_name = "Dpngtm/
|
11 |
model = Wav2Vec2ForSequenceClassification.from_pretrained(model_name)
|
12 |
processor = Wav2Vec2Processor.from_pretrained(model_name, num_labels=len(emotion_labels))
|
13 |
|
|
|
7 |
emotion_labels = ["angry", "calm", "disgust", "fearful", "happy", "neutral", "sad", "surprised"]
|
8 |
|
9 |
# Load model and processor
|
10 |
+
model_name = "Dpngtm/wav2vec2-emotion-recognition" # Replace with your model's Hugging Face Hub path
|
11 |
model = Wav2Vec2ForSequenceClassification.from_pretrained(model_name)
|
12 |
processor = Wav2Vec2Processor.from_pretrained(model_name, num_labels=len(emotion_labels))
|
13 |
|