Spaces:
Running
Running
Update app.py
Browse files
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():
|