TiberiuCristianLeon commited on
Commit
f039ca9
·
verified ·
1 Parent(s): aff7959

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -43,7 +43,7 @@ models = ["Helsinki-NLP", "QUICKMT", "Argos", "Lego-MT/Lego-MT", "HPLT", "HPLT-O
43
  "bigscience/bloomz-560m", "bigscience/bloomz-1b1", "bigscience/bloomz-1b7", "bigscience/bloomz-3b",
44
  "t5-small", "t5-base", "t5-large",
45
  "google/flan-t5-small", "google/flan-t5-base", "google/flan-t5-large", "google/flan-t5-xl",
46
- "google/madlad400-3b-mt", "Heng666/madlad400-3b-mt-ct2-int8", "Heng666/madlad400-7b-mt-ct2-int8",
47
  "utter-project/EuroLLM-1.7B", "utter-project/EuroLLM-1.7B-Instruct",
48
  "Unbabel/Tower-Plus-2B", "Unbabel/TowerInstruct-7B-v0.2", "Unbabel/TowerInstruct-Mistral-7B-v0.2",
49
  "HuggingFaceTB/SmolLM3-3B", "winninghealth/WiNGPT-Babel-2",
@@ -589,7 +589,7 @@ def translate_text(model_name: str, s_language: str, t_language: str, input_text
589
  elif 'flan' in model_name.lower():
590
  translated_text = Translators(model_name, s_language, t_language, input_text).flan()
591
 
592
- elif 'mt-ct2-int8' in model_name.lower():
593
  translated_text = Translators(model_name, sl, tl, input_text).madladct2()
594
 
595
  elif 'madlad' in model_name.lower():
 
43
  "bigscience/bloomz-560m", "bigscience/bloomz-1b1", "bigscience/bloomz-1b7", "bigscience/bloomz-3b",
44
  "t5-small", "t5-base", "t5-large",
45
  "google/flan-t5-small", "google/flan-t5-base", "google/flan-t5-large", "google/flan-t5-xl",
46
+ "google/madlad400-3b-mt", "Heng666/madlad400-3b-mt-ct2", "Heng666/madlad400-3b-mt-ct2-int8", "Heng666/madlad400-7b-mt-ct2-int8",
47
  "utter-project/EuroLLM-1.7B", "utter-project/EuroLLM-1.7B-Instruct",
48
  "Unbabel/Tower-Plus-2B", "Unbabel/TowerInstruct-7B-v0.2", "Unbabel/TowerInstruct-Mistral-7B-v0.2",
49
  "HuggingFaceTB/SmolLM3-3B", "winninghealth/WiNGPT-Babel-2",
 
589
  elif 'flan' in model_name.lower():
590
  translated_text = Translators(model_name, s_language, t_language, input_text).flan()
591
 
592
+ elif 'mt-ct2' in model_name.lower():
593
  translated_text = Translators(model_name, sl, tl, input_text).madladct2()
594
 
595
  elif 'madlad' in model_name.lower():