Update app.py

app.py

@@ -8,11 +8,11 @@ import httpx
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 # Language options and mappings
 favourite_langs = {"Romanian": "ro", "German": "de", "English": "en", "-----": "-----"}
-# langs = ["German", "Romanian", "English", "French", "Spanish", "Italian",]
 df = pl.read_parquet("isolanguages.parquet")
 non_empty_isos = df.slice(1).filter(pl.col("ISO639-1") != "").rows()
 all_langs = {iso[0]: (iso[1], iso[2], iso[3]) for iso in non_empty_isos} # {'Romanian': ('ro', 'rum', 'ron')}
 name_to_iso1 = {iso[0]: iso[1] for iso in non_empty_isos} # {'Romanian': 'ro', 'German': 'de'}
+# langs = ["German", "Romanian", "English", "French", "Spanish", "Italian"]
 langs = list(favourite_langs.keys())
 langs.extend(list(all_langs.keys())) # Language options as list, add favourite languages first
 # all_langs = languagecodes.iso_languages_byname
@@ -38,19 +38,6 @@ models = ["Helsinki-NLP", "QUICKMT", "Argos", "Lego-MT/Lego-MT", "HPLT", "HPLT-O
          "tencent/Hunyuan-MT-7B",
          "openGPT-X/Teuken-7B-instruct-commercial-v0.4", "openGPT-X/Teuken-7B-instruct-v0.6",
          ]
-allmodels = ["Helsinki-NLP",
-          "Helsinki-NLP/opus-mt-tc-bible-big-mul-mul", "Helsinki-NLP/opus-mt-tc-bible-big-mul-deu_eng_nld",
-          "Helsinki-NLP/opus-mt-tc-bible-big-mul-deu_eng_fra_por_spa", "Helsinki-NLP/opus-mt-tc-bible-big-deu_eng_fra_por_spa-mul",
-          "Helsinki-NLP/opus-mt-tc-bible-big-roa-deu_eng_fra_por_spa", "Helsinki-NLP/opus-mt-tc-bible-big-deu_eng_fra_por_spa-roa",
-          "facebook/nllb-200-distilled-600M", "facebook/nllb-200-distilled-1.3B", "facebook/nllb-200-1.3B", "facebook/nllb-200-3.3B",
-          "facebook/mbart-large-50-many-to-many-mmt", "facebook/mbart-large-50-one-to-many-mmt", "facebook/mbart-large-50-many-to-one-mmt",
-          "facebook/m2m100_418M", "facebook/m2m100_1.2B", "Lego-MT/Lego-MT",
-          "bigscience/mt0-small", "bigscience/mt0-base", "bigscience/mt0-large", "bigscience/mt0-xl",
-          "bigscience/bloomz-560m", "bigscience/bloomz-1b1", "bigscience/bloomz-1b7", "bigscience/bloomz-3b",
-          "t5-small", "t5-base", "t5-large",
-          "google/flan-t5-small", "google/flan-t5-base", "google/flan-t5-large", "google/flan-t5-xl",
-          "google/madlad400-3b-mt", "jbochi/madlad400-3b-mt",
-          ]
 class Translators:
     def __init__(self, model_name: str, sl: str, tl: str, input_text: str):
         self.model_name = model_name
@@ -539,7 +526,7 @@ def translate_text(model_name: str, s_language: str, t_language: str, input_text
     message_text = f'Translated from {s_language} to {t_language} with {model_name}'
     translated_text = None
     try:
-        if in model_name == "Helsinki-NLP/opus-mt-tc-bible-big-roa-en":
+        if model_name == "Helsinki-NLP/opus-mt-tc-bible-big-roa-en":
             translated_text, message_text = Translators(model_name, sl, tl, input_text).simplepipe()
         
         elif "-mul" in model_name.lower() or "mul-" in model_name.lower() or "-roa" in model_name.lower():
@@ -548,6 +535,9 @@ def translate_text(model_name: str, s_language: str, t_language: str, input_text
         elif model_name == "Helsinki-NLP":
             translated_text, message_text = Translators(model_name, sl, tl, input_text).HelsinkiNLP()
 
+        elif model_name == "QUICKMT":
+            translated_text, message_text = Translators(model_name, sl, tl, input_text).quickmt()
+
         elif "HPLT" in model_name:
             if model_name == "HPLT-OPUS":
                 translated_text, message = Translators(model_name, sl, tl, input_text).hplt(opus = True)
@@ -618,9 +608,6 @@ def translate_text(model_name: str, s_language: str, t_language: str, input_text
         elif model_name == "Bergamot":
             translated_text, message_text = Translators(model_name, s_language, t_language, input_text).bergamot()
 
-        elif model_name == "QUICKMT":
-            translated_text, message_text = Translators(model_name, sl, tl, input_text).quickmt()
-
         elif "Hunyuan" in model_name:      
             translated_text = Translators(model_name, s_language, t_language, input_text).hunyuan()