Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ from transformers import T5Tokenizer, T5ForConditionalGeneration, AutoTokenizer,
|
|
4 |
|
5 |
@spaces.GPU
|
6 |
def translate_text(input_text, sselected_language, tselected_language, model_name):
|
7 |
-
langs = {"English": "en", "Romanian": "ro", "German": "de", "French": "fr", "Spanish": "es",
|
8 |
sl = langs[sselected_language]
|
9 |
tl = langs[tselected_language]
|
10 |
|
@@ -38,7 +38,7 @@ def translate_text(input_text, sselected_language, tselected_language, model_nam
|
|
38 |
|
39 |
return translated_text
|
40 |
|
41 |
-
options = ["German", "Romanian", "English", "French", "Spanish"]
|
42 |
models = ["Helsinki-NLP", "t5-base", "t5-small", "t5-large"]
|
43 |
|
44 |
# Define a function to swap dropdown values
|
|
|
4 |
|
5 |
@spaces.GPU
|
6 |
def translate_text(input_text, sselected_language, tselected_language, model_name):
|
7 |
+
langs = {"English": "en", "Romanian": "ro", "German": "de", "French": "fr", "Spanish": "es", "Italian": "it"}
|
8 |
sl = langs[sselected_language]
|
9 |
tl = langs[tselected_language]
|
10 |
|
|
|
38 |
|
39 |
return translated_text
|
40 |
|
41 |
+
options = ["German", "Romanian", "English", "French", "Spanish", "Italian"]
|
42 |
models = ["Helsinki-NLP", "t5-base", "t5-small", "t5-large"]
|
43 |
|
44 |
# Define a function to swap dropdown values
|