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 =
|
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):
|
|
|
7 |
favourite_langs = {"German": "de", "Romanian": "ro", "English": "en", "-----": "-----"}
|
8 |
all_langs = languagecodes.iso_languages
|
9 |
|
10 |
+
options = list(favourite_langs.keys())
|
11 |
+
options.extend(list(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):
|