TiberiuCristianLeon commited on
Commit
d836ba8
·
verified ·
1 Parent(s): 1810ebb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -27,7 +27,7 @@ models = ["Helsinki-NLP", "QUICKMT", "Argos", "Google", "HPLT", "HPLT-OPUS",
27
  "facebook/m2m100_418M", "facebook/m2m100_1.2B", "Lego-MT/Lego-MT",
28
  "bigscience/mt0-small", "bigscience/mt0-base", "bigscience/mt0-large", "bigscience/mt0-xl",
29
  "bigscience/bloomz-560m", "bigscience/bloomz-1b1", "bigscience/bloomz-1b7", "bigscience/bloomz-3b",
30
- "t5-small", "t5-base", "t5-large",
31
  "google/flan-t5-small", "google/flan-t5-base", "google/flan-t5-large", "google/flan-t5-xl",
32
  "google/madlad400-3b-mt", "jbochi/madlad400-3b-mt",
33
  "HuggingFaceTB/SmolLM3-3B", "winninghealth/WiNGPT-Babel-2",
@@ -494,7 +494,7 @@ def translate_text(input_text: str, s_language: str, t_language: str, model_name
494
  elif "lego" in model_name.lower():
495
  translated_text = Translators(model_name, sl, tl, input_text).LegoMT()
496
 
497
- elif model_name.startswith('t5'):
498
  translated_text = Translators(model_name, s_language, t_language, input_text).tfive()
499
 
500
  elif 'flan' in model_name.lower():
 
27
  "facebook/m2m100_418M", "facebook/m2m100_1.2B", "Lego-MT/Lego-MT",
28
  "bigscience/mt0-small", "bigscience/mt0-base", "bigscience/mt0-large", "bigscience/mt0-xl",
29
  "bigscience/bloomz-560m", "bigscience/bloomz-1b1", "bigscience/bloomz-1b7", "bigscience/bloomz-3b",
30
+ "google-t5/t5-small", "google-t5/t5-base", "google-t5/t5-large",
31
  "google/flan-t5-small", "google/flan-t5-base", "google/flan-t5-large", "google/flan-t5-xl",
32
  "google/madlad400-3b-mt", "jbochi/madlad400-3b-mt",
33
  "HuggingFaceTB/SmolLM3-3B", "winninghealth/WiNGPT-Babel-2",
 
494
  elif "lego" in model_name.lower():
495
  translated_text = Translators(model_name, sl, tl, input_text).LegoMT()
496
 
497
+ elif model_name.startswith('google-t5'):
498
  translated_text = Translators(model_name, s_language, t_language, input_text).tfive()
499
 
500
  elif 'flan' in model_name.lower():