3loi commited on
Commit
bd1d60b
1 Parent(s): 76fe58b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -8,10 +8,9 @@ import numpy as np
8
 
9
 
10
 
11
- mean, std = -8.278621631819787e-05, 0.08485510250851999
12
- model_sr=model.config.sampling_rate
13
 
14
- id2label = {0: 'arousal', 1: 'dominance', 2: 'valence'}
 
15
  description_text = "Multi-label (arousal, dominance, valence) Odyssey 2024 Emotion Recognition competition baseline model.<br> \
16
  The model is trained on MSP-Podcast. \
17
  For more details visit: [HuggingFace](https://huggingface.co/3loi/SER-Odyssey-Baseline-WavLM-Multi-Attributes), \
@@ -21,7 +20,9 @@ description_text = "Multi-label (arousal, dominance, valence) Odyssey 2024 Emoti
21
 
22
  def classify_audio(audio_file):
23
  model = AutoModelForAudioClassification.from_pretrained("3loi/SER-Odyssey-Baseline-WavLM-Multi-Attributes", trust_remote_code=True)
24
-
 
 
25
 
26
  sr, raw_wav = audio_file
27
 
 
8
 
9
 
10
 
 
 
11
 
12
+
13
+
14
  description_text = "Multi-label (arousal, dominance, valence) Odyssey 2024 Emotion Recognition competition baseline model.<br> \
15
  The model is trained on MSP-Podcast. \
16
  For more details visit: [HuggingFace](https://huggingface.co/3loi/SER-Odyssey-Baseline-WavLM-Multi-Attributes), \
 
20
 
21
  def classify_audio(audio_file):
22
  model = AutoModelForAudioClassification.from_pretrained("3loi/SER-Odyssey-Baseline-WavLM-Multi-Attributes", trust_remote_code=True)
23
+ mean, std = model.config.mean, model.config.std
24
+ model_sr = model.config.sampling_rate
25
+ id2label = model.config.id2label
26
 
27
  sr, raw_wav = audio_file
28