Hugging Face's logo

Spaces:
ameliabb0913
/
AudioEmotionRecognitionTask
Running

App Files Community
ameliabb0913 commited on
Commit
2f516ad
·
verified ·
1 Parent(s): bb12bfd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from transformers import Wav2Vec2FeatureExtractor, Wav2Vec2ForSequenceClassifica
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
 
 
5
 
6
  MODEL_NAME = "ameliabb0913/emotion-classifier"
7
  processor = Wav2Vec2FeatureExtractor.from_pretrained(MODEL_NAME)
8
+ model = Wav2Vec2ForSequenceClassification.from_pretrained(MODEL_NAME, from_safetensors=True)
9
  model.eval()
10
 
11