Update app.py
Browse files
app.py
CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
|
|
2 |
import spaces
|
3 |
from transformers import T5Tokenizer, T5ForConditionalGeneration, AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
|
4 |
|
5 |
-
langs = {"
|
6 |
options = list(langs.keys())
|
7 |
models = ["Helsinki-NLP", "t5-base", "t5-small", "t5-large", "facebook/nllb-200-distilled-1.3B", "facebook/nllb-200-distilled-600M"]
|
8 |
|
@@ -55,8 +55,8 @@ def create_interface():
|
|
55 |
input_text = gr.Textbox(label="Enter text to translate:", placeholder="Type your text here...")
|
56 |
|
57 |
with gr.Row():
|
58 |
-
sselected_language = gr.Dropdown(choices=options, label="Source language")
|
59 |
-
tselected_language = gr.Dropdown(choices=options, label="Target language")
|
60 |
swap_button = gr.Button("Swap Languages")
|
61 |
swap_button.click(fn=swap_languages, inputs=[sselected_language, tselected_language], outputs=[sselected_language, tselected_language])
|
62 |
|
|
|
2 |
import spaces
|
3 |
from transformers import T5Tokenizer, T5ForConditionalGeneration, AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
|
4 |
|
5 |
+
langs = {"German": ("de", "deu_Latn"), "Romanian": ("ro", "ron_Latn"), "English": ("en", "eng_Latn"), "French": ("fr", "fra_Latn"), "Spanish": ("es", "spa_Latn"), "Italian": ("it", "ita_Latn")}
|
6 |
options = list(langs.keys())
|
7 |
models = ["Helsinki-NLP", "t5-base", "t5-small", "t5-large", "facebook/nllb-200-distilled-1.3B", "facebook/nllb-200-distilled-600M"]
|
8 |
|
|
|
55 |
input_text = gr.Textbox(label="Enter text to translate:", placeholder="Type your text here...")
|
56 |
|
57 |
with gr.Row():
|
58 |
+
sselected_language = gr.Dropdown(choices=options, value = options[0], label="Source language" interactive=True)
|
59 |
+
tselected_language = gr.Dropdown(choices=options, value = options[1], label="Target language", interactive=True)
|
60 |
swap_button = gr.Button("Swap Languages")
|
61 |
swap_button.click(fn=swap_languages, inputs=[sselected_language, tselected_language], outputs=[sselected_language, tselected_language])
|
62 |
|