TiberiuCristianLeon commited on
Commit
4596a5d
·
verified ·
1 Parent(s): 6dc437d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
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
- langs = languagecodes.iso_languages
9
 
10
- # options = list(langs.keys())
11
- options = [(k, v) for k,v in favourite_langs.items()].extend([(k, v) for k,v in 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,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 = langs[sselected_language]
25
- tl = 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:
 
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: