Update app.py
Browse files
app.py
CHANGED
@@ -1,14 +1,14 @@
|
|
1 |
import gradio as gr
|
2 |
import torch
|
3 |
-
from transformers import Wav2Vec2ForSequenceClassification, Wav2Vec2Processor
|
4 |
import librosa
|
|
|
5 |
|
6 |
-
# Placeholder model (Replace later with your trained model)
|
7 |
MODEL_NAME = "ameliabb0913/emotion-classifier"
|
8 |
-
processor =
|
9 |
model = Wav2Vec2ForSequenceClassification.from_pretrained(MODEL_NAME)
|
10 |
model.eval()
|
11 |
|
|
|
12 |
# Emotion labels (based on the dataset used to train the model)
|
13 |
id2label = {
|
14 |
0: "Neutral",
|
|
|
1 |
import gradio as gr
|
2 |
import torch
|
|
|
3 |
import librosa
|
4 |
+
from transformers import Wav2Vec2FeatureExtractor, Wav2Vec2ForSequenceClassification
|
5 |
|
|
|
6 |
MODEL_NAME = "ameliabb0913/emotion-classifier"
|
7 |
+
processor = Wav2Vec2FeatureExtractor.from_pretrained(MODEL_NAME)
|
8 |
model = Wav2Vec2ForSequenceClassification.from_pretrained(MODEL_NAME)
|
9 |
model.eval()
|
10 |
|
11 |
+
|
12 |
# Emotion labels (based on the dataset used to train the model)
|
13 |
id2label = {
|
14 |
0: "Neutral",
|