Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ import torch
|
|
3 |
import librosa
|
4 |
from transformers import Wav2Vec2FeatureExtractor, Wav2Vec2ForSequenceClassification
|
5 |
|
6 |
-
MODEL_NAME = "ameliabb0913/emotion-
|
7 |
processor = Wav2Vec2FeatureExtractor.from_pretrained(MODEL_NAME)
|
8 |
model = Wav2Vec2ForSequenceClassification.from_pretrained(MODEL_NAME, trust_remote_code=True)
|
9 |
model.eval()
|
|
|
3 |
import librosa
|
4 |
from transformers import Wav2Vec2FeatureExtractor, Wav2Vec2ForSequenceClassification
|
5 |
|
6 |
+
MODEL_NAME = "ameliabb0913/emotion-classifier1"
|
7 |
processor = Wav2Vec2FeatureExtractor.from_pretrained(MODEL_NAME)
|
8 |
model = Wav2Vec2ForSequenceClassification.from_pretrained(MODEL_NAME, trust_remote_code=True)
|
9 |
model.eval()
|