3loi commited on
Commit
5642657
1 Parent(s): ff9ea0c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -10,7 +10,11 @@ import numpy as np
10
 
11
  mean, std = -8.278621631819787e-05, 0.08485510250851999
12
  id2label = {0: 'arousal', 1: 'dominance', 2: 'valence'}
13
-
 
 
 
 
14
 
15
 
16
  def classify_audio(audio_file):
@@ -47,8 +51,7 @@ def main():
47
 
48
  iface = gr.Interface(fn=classify_audio, inputs=gr.Audio(sources=["upload", "microphone"], label="Audio file"),
49
  outputs=gr.Text(), title="Speech Emotion Recognition App",
50
- description="Upload an audio file and hit the 'Submit'\
51
- button")
52
 
53
  iface.launch()
54
 
 
10
 
11
  mean, std = -8.278621631819787e-05, 0.08485510250851999
12
  id2label = {0: 'arousal', 1: 'dominance', 2: 'valence'}
13
+ description_text = "Multi-label (arousal, dominance, valence) Odyssey 2024 Emotion Recognition competition baseline model.<br> \
14
+ The model is trained on MSP-Podcast. \
15
+ For more details visit: [HuggingFace](https://huggingface.co/3loi/SER-Odyssey-Baseline-WavLM-Multi-Attributes), \
16
+ [paper/soon]() and [GitHub](https://github.com/MSP-UTD/MSP-Podcast_Challenge/tree/main). <br><br>\
17
+ Upload an audio file and hit the 'Submit' button to predict the emotion"
18
 
19
 
20
  def classify_audio(audio_file):
 
51
 
52
  iface = gr.Interface(fn=classify_audio, inputs=gr.Audio(sources=["upload", "microphone"], label="Audio file"),
53
  outputs=gr.Text(), title="Speech Emotion Recognition App",
54
+ description=description_text)
 
55
 
56
  iface.launch()
57