lpw commited on
Commit
ed7aac3
1 Parent(s): b6b1a72

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -7,8 +7,8 @@ from audio_pipe import SpeechToSpeechPipeline
7
  # io2 = gr.Interface.load("huggingface/facebook/xm_transformer_s2ut_hk-en", api_key=os.environ['api_key'])
8
  # io3 = gr.Interface.load("huggingface/facebook/xm_transformer_unity_en-hk", api_key=os.environ['api_key'])
9
  # io4 = gr.Interface.load("huggingface/facebook/xm_transformer_unity_hk-en", api_key=os.environ['api_key'])
10
- pipe1 = SpeechToSpeechPipeline("facebook/xm_transformer_s2ut_en-hk")
11
- pipe2 = SpeechToSpeechPipeline("facebook/xm_transformer_s2ut_hk-en")
12
  pipe3 = SpeechToSpeechPipeline("facebook/xm_transformer_unity_en-hk")
13
  pipe4 = SpeechToSpeechPipeline("facebook/xm_transformer_unity_hk-en")
14
 
@@ -132,7 +132,8 @@ with block:
132
  )
133
 
134
  btn = gr.Button("Submit")
135
- model = gr.Dropdown(choices=["xm_transformer_unity_en-hk", "xm_transformer_unity_hk-en", "xm_transformer_unity_hk-en_gpu", "xm_transformer_s2ut_en-hk", "xm_transformer_s2ut_hk-en"], value="xm_transformer_unity_en-hk",type="value", label="Model")
 
136
  out = gr.Audio(label="Output")
137
 
138
  btn.click(inference, inputs=[audio, model], outputs=out)
 
7
  # io2 = gr.Interface.load("huggingface/facebook/xm_transformer_s2ut_hk-en", api_key=os.environ['api_key'])
8
  # io3 = gr.Interface.load("huggingface/facebook/xm_transformer_unity_en-hk", api_key=os.environ['api_key'])
9
  # io4 = gr.Interface.load("huggingface/facebook/xm_transformer_unity_hk-en", api_key=os.environ['api_key'])
10
+ # pipe1 = SpeechToSpeechPipeline("facebook/xm_transformer_s2ut_en-hk")
11
+ # pipe2 = SpeechToSpeechPipeline("facebook/xm_transformer_s2ut_hk-en")
12
  pipe3 = SpeechToSpeechPipeline("facebook/xm_transformer_unity_en-hk")
13
  pipe4 = SpeechToSpeechPipeline("facebook/xm_transformer_unity_hk-en")
14
 
 
132
  )
133
 
134
  btn = gr.Button("Submit")
135
+ # model = gr.Dropdown(choices=["xm_transformer_unity_en-hk", "xm_transformer_unity_hk-en", "xm_transformer_s2ut_en-hk", "xm_transformer_s2ut_hk-en"], value="xm_transformer_unity_en-hk",type="value", label="Model")
136
+ model = gr.Dropdown(choices=["xm_transformer_unity_en-hk", "xm_transformer_unity_hk-en"], value="xm_transformer_unity_en-hk",type="value", label="Model")
137
  out = gr.Audio(label="Output")
138
 
139
  btn.click(inference, inputs=[audio, model], outputs=out)