Spaces:
Running
Running
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):
|