TiberiuCristianLeon commited on
Commit
2da2e03
·
verified ·
1 Parent(s): 7b79ed9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 = [(k for k in favourite_langs.keys()]
11
- options.extend([k for k in 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):
 
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):