Update app.py
Browse files
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
|
11 |
-
options.extend([
|
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]
|
89 |
-
tselected_language = gr.Dropdown(choices=options, value = options[1]
|
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 |
|