juulaii commited on
Commit
076cb4c
·
1 Parent(s): 802ffcf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -12
app.py CHANGED
@@ -2,21 +2,21 @@ import gradio as gr
2
 
3
  #Get models
4
  #ASR model for input speech
5
- ui = gr.Interface.load("huggingface/facebook/hubert-large-ls960-ft",
6
- inputs=gr.inputs.Audio(label="Record Audio", type="filepath", source = "microphone"))
7
 
8
- #translates English to Spanish text
9
- #translator = gr.Interface.load("huggingface/Helsinki-NLP/opus-mt-en-es",
10
- # outputs=gr.outputs.Textbox(label="English to Spanish Translated Text"))
11
- #TTS model for output speech
12
- #text2speech = gr.Interface.load("huggingface/facebook/tts_transformer-es-css10",
13
- # outputs=gr.outputs.Audio(label="English to Spanish Translated Audio"),
14
- # allow_flagging="never")
15
 
16
 
17
- #ui = gr.Series(speech2text, translator) #outputs Spanish text translation
18
- #en2es = gr.Series(translate, text2speech) #outputs Spanish audio
19
- #ui = gr.Parallel(translate, en2es) #allows transcription of Spanish audio
20
 
21
  #gradio interface
22
  ui.title = "English to Spanish Speech Translator"
 
2
 
3
  #Get models
4
  #ASR model for input speech
5
+ speech2text = gr.Interface.load("huggingface/facebook/hubert-large-ls960-ft",
6
+ inputs=gr.inputs.Audio(label="Upload Audio", type="filepath", source = "upload"))
7
 
8
+ translates English to Spanish text
9
+ translator = gr.Interface.load("huggingface/Helsinki-NLP/opus-mt-en-es",
10
+ outputs=gr.outputs.Textbox(label="English to Spanish Translated Text"))
11
+ TTS model for output speech
12
+ text2speech = gr.Interface.load("huggingface/facebook/tts_transformer-es-css10",
13
+ outputs=gr.outputs.Audio(label="English to Spanish Translated Audio"),
14
+ allow_flagging="never")
15
 
16
 
17
+ translate = gr.Series(speech2text, translator) #outputs Spanish text translation
18
+ en2es = gr.Series(translate, text2speech) #outputs Spanish audio
19
+ ui = gr.Parallel(translate, en2es) #allows transcription of Spanish audio
20
 
21
  #gradio interface
22
  ui.title = "English to Spanish Speech Translator"