Update app.py
Browse files
app.py
CHANGED
@@ -5,10 +5,10 @@ from transformers import T5Tokenizer, T5ForConditionalGeneration, AutoTokenizer,
|
|
5 |
import languagecodes
|
6 |
|
7 |
favourite_langs = {"German": "de", "Romanian": "ro", "English": "en", "-----": "-----"}
|
8 |
-
|
9 |
|
10 |
-
|
11 |
-
options
|
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):
|
@@ -21,8 +21,8 @@ def model_to_cuda(model):
|
|
21 |
return model
|
22 |
@spaces.GPU
|
23 |
def translate_text(input_text, sselected_language, tselected_language, model_name):
|
24 |
-
sl =
|
25 |
-
tl =
|
26 |
message_text = f'Translated from {sselected_language} to {tselected_language} with {model_name}'
|
27 |
if model_name == "Helsinki-NLP":
|
28 |
try:
|
|
|
5 |
import languagecodes
|
6 |
|
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):
|
|
|
21 |
return model
|
22 |
@spaces.GPU
|
23 |
def translate_text(input_text, sselected_language, tselected_language, model_name):
|
24 |
+
sl = all_langs[sselected_language]
|
25 |
+
tl = all_langs[tselected_language]
|
26 |
message_text = f'Translated from {sselected_language} to {tselected_language} with {model_name}'
|
27 |
if model_name == "Helsinki-NLP":
|
28 |
try:
|