jthteo commited on
Commit
5457f98
1 Parent(s): bf762d3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -7,9 +7,9 @@ io1 = gr.Interface.load("huggingface/facebook/xm_transformer_s2ut_en-hk")
7
  io2 = gr.Interface.load("huggingface/facebook/xm_transformer_s2ut_hk-en")
8
 
9
  def inference(audio, model):
10
- if model == "hokkientranslator_en_hk":
11
  out_audio = io1(audio)
12
- elif model == "hokkientranslator_hk_en":
13
  out_audio = io2(audio)
14
  return out_audio
15
 
@@ -122,7 +122,7 @@ with block:
122
  )
123
 
124
  btn = gr.Button("Submit")
125
- model = gr.Dropdown(choices=["hokkientranslator_en_hk", "huggingface/facebook/xm_transformer_s2ut_hk-en"], value="hokkientranslator_en_hk",type="value", label="Model")
126
  out = gr.Audio(label="Output")
127
 
128
  btn.click(inference, inputs=[audio, model], outputs=out)
 
7
  io2 = gr.Interface.load("huggingface/facebook/xm_transformer_s2ut_hk-en")
8
 
9
  def inference(audio, model):
10
+ if model == "english->hokkien":
11
  out_audio = io1(audio)
12
+ elif model == "hokkien->english":
13
  out_audio = io2(audio)
14
  return out_audio
15
 
 
122
  )
123
 
124
  btn = gr.Button("Submit")
125
+ model = gr.Dropdown(choices=["english->hokkien", "hokkien->english"], value="english->hokkien",type="value", label="Model")
126
  out = gr.Audio(label="Output")
127
 
128
  btn.click(inference, inputs=[audio, model], outputs=out)