hysts HF staff commited on
Commit
c3b9be5
1 Parent(s): 748b9ed

Fix target language list

Browse files
Files changed (2) hide show
  1. app.py +4 -2
  2. lang_list.py +3 -2
app.py CHANGED
@@ -11,9 +11,11 @@ from huggingface_hub import snapshot_download
11
  from seamless_communication.inference import Translator
12
 
13
  from lang_list import (
 
14
  LANGUAGE_NAME_TO_CODE,
15
  S2ST_TARGET_LANGUAGE_NAMES,
16
  S2TT_TARGET_LANGUAGE_NAMES,
 
17
  T2TT_TARGET_LANGUAGE_NAMES,
18
  TEXT_SOURCE_LANGUAGE_NAMES,
19
  )
@@ -201,7 +203,7 @@ with gr.Blocks() as demo_t2st:
201
  )
202
  target_language = gr.Dropdown(
203
  label="Target language",
204
- choices=T2TT_TARGET_LANGUAGE_NAMES,
205
  value=DEFAULT_TARGET_LANGUAGE,
206
  )
207
  input_text = gr.Textbox(label="Input text")
@@ -311,7 +313,7 @@ with gr.Blocks() as demo_asr:
311
  with gr.Group():
312
  target_language = gr.Dropdown(
313
  label="Target language",
314
- choices=S2ST_TARGET_LANGUAGE_NAMES,
315
  value=DEFAULT_TARGET_LANGUAGE,
316
  )
317
  input_audio = gr.Audio(label="Input speech", type="filepath")
 
11
  from seamless_communication.inference import Translator
12
 
13
  from lang_list import (
14
+ ASR_TARGET_LANGUAGE_NAMES,
15
  LANGUAGE_NAME_TO_CODE,
16
  S2ST_TARGET_LANGUAGE_NAMES,
17
  S2TT_TARGET_LANGUAGE_NAMES,
18
+ T2ST_TARGET_LANGUAGE_NAMES,
19
  T2TT_TARGET_LANGUAGE_NAMES,
20
  TEXT_SOURCE_LANGUAGE_NAMES,
21
  )
 
203
  )
204
  target_language = gr.Dropdown(
205
  label="Target language",
206
+ choices=T2ST_TARGET_LANGUAGE_NAMES,
207
  value=DEFAULT_TARGET_LANGUAGE,
208
  )
209
  input_text = gr.Textbox(label="Input text")
 
313
  with gr.Group():
314
  target_language = gr.Dropdown(
315
  label="Target language",
316
+ choices=ASR_TARGET_LANGUAGE_NAMES,
317
  value=DEFAULT_TARGET_LANGUAGE,
318
  )
319
  input_audio = gr.Audio(label="Input speech", type="filepath")
lang_list.py CHANGED
@@ -247,8 +247,9 @@ s2st_target_language_codes = [
247
  "vie",
248
  ]
249
  S2ST_TARGET_LANGUAGE_NAMES = sorted([language_code_to_name[code] for code in s2st_target_language_codes])
 
250
 
251
- # S2TT / ASR
252
  S2TT_TARGET_LANGUAGE_NAMES = TEXT_SOURCE_LANGUAGE_NAMES
253
- # T2TT
254
  T2TT_TARGET_LANGUAGE_NAMES = TEXT_SOURCE_LANGUAGE_NAMES
 
 
247
  "vie",
248
  ]
249
  S2ST_TARGET_LANGUAGE_NAMES = sorted([language_code_to_name[code] for code in s2st_target_language_codes])
250
+ T2ST_TARGET_LANGUAGE_NAMES = S2ST_TARGET_LANGUAGE_NAMES
251
 
252
+ # S2TT / T2TT / ASR
253
  S2TT_TARGET_LANGUAGE_NAMES = TEXT_SOURCE_LANGUAGE_NAMES
 
254
  T2TT_TARGET_LANGUAGE_NAMES = TEXT_SOURCE_LANGUAGE_NAMES
255
+ ASR_TARGET_LANGUAGE_NAMES = TEXT_SOURCE_LANGUAGE_NAMES