juulaii commited on
Commit
c52add3
1 Parent(s): 933b458

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -9
app.py CHANGED
@@ -2,21 +2,21 @@ import gradio as gr
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="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
- 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"
 
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"