TiberiuCristianLeon commited on
Commit
7b79ed9
·
verified ·
1 Parent(s): 3137920

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -7,8 +7,8 @@ import languagecodes
7
  favourite_langs = {"German": "de", "Romanian": "ro", "English": "en", "-----": "-----"}
8
  all_langs = languagecodes.iso_languages
9
 
10
- options = [(k, v) for k,v in favourite_langs.items()]
11
- options.extend([(k, v) for k,v in all_langs.items()])
12
  models = ["Helsinki-NLP", "t5-base", "t5-small", "t5-large", "facebook/nllb-200-distilled-600M", "facebook/nllb-200-distilled-1.3B", "facebook/mbart-large-50-many-to-many-mmt"]
13
 
14
  def model_to_cuda(model):
@@ -85,8 +85,8 @@ def create_interface():
85
  input_text = gr.Textbox(label="Enter text to translate:", placeholder="Type your text here...")
86
 
87
  with gr.Row():
88
- sselected_language = gr.Dropdown(choices=options, value = options[0][0], label="Source language", interactive=True)
89
- tselected_language = gr.Dropdown(choices=options, value = options[1][0], label="Target language", interactive=True)
90
  swap_button = gr.Button("Swap Languages")
91
  swap_button.click(fn=swap_languages, inputs=[sselected_language, tselected_language], outputs=[sselected_language, tselected_language])
92
 
 
7
  favourite_langs = {"German": "de", "Romanian": "ro", "English": "en", "-----": "-----"}
8
  all_langs = languagecodes.iso_languages
9
 
10
+ options = [(k for k in favourite_langs.keys()]
11
+ options.extend([k for k in all_langs.keys()])
12
  models = ["Helsinki-NLP", "t5-base", "t5-small", "t5-large", "facebook/nllb-200-distilled-600M", "facebook/nllb-200-distilled-1.3B", "facebook/mbart-large-50-many-to-many-mmt"]
13
 
14
  def model_to_cuda(model):
 
85
  input_text = gr.Textbox(label="Enter text to translate:", placeholder="Type your text here...")
86
 
87
  with gr.Row():
88
+ sselected_language = gr.Dropdown(choices=options, value = options[0], label="Source language", interactive=True)
89
+ tselected_language = gr.Dropdown(choices=options, value = options[1], label="Target language", interactive=True)
90
  swap_button = gr.Button("Swap Languages")
91
  swap_button.click(fn=swap_languages, inputs=[sselected_language, tselected_language], outputs=[sselected_language, tselected_language])
92