lorenzoscottb commited on
Commit
d757f0d
Β·
1 Parent(s): d0ebdbe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -2,12 +2,12 @@
2
  import gradio as gr
3
  from gradio.mix import Series
4
 
5
- description = "This space mixes DReAMy's large-multilingual [emotion-classification model](https://github.com/lorenzoscottb/DReAMy) with whisper-base from OpenAI, to classify recordings on their emotional content accordingly to the Hall and Van de Castle framework. For more details, see the [Bertolini et al., 23](https://arxiv.org/abs/2302.14828v1) pre-print."
6
  title = "DReAM v. II"
7
 
8
- whisper = gr.Interface.load("models/openai/whisper-base", device="cpu")
9
 
10
- interface_model_L = gr.Interface.load("models/DReAMy-lib/xlm-roberta-large-DreamBank-emotion-presence", device="cpu")
11
 
12
  Series(
13
  whisper,
 
2
  import gradio as gr
3
  from gradio.mix import Series
4
 
5
+ description = "This space mixes DReAMy's large-multilingual [emotion-classification model](https://github.com/lorenzoscottb/DReAMy) with Whisper from OpenAI, to classify recordings on their emotional content accordingly to the Hall and Van de Castle framework. For more details, see the [Bertolini et al., 23](https://arxiv.org/abs/2302.14828v1) pre-print."
6
  title = "DReAM v. II"
7
 
8
+ whisper = gr.Interface.load("models/openai/whisper-large")
9
 
10
+ interface_model_L = gr.Interface.load("models/DReAMy-lib/xlm-roberta-large-DreamBank-emotion-presence")
11
 
12
  Series(
13
  whisper,