Update app.py

app.py

@@ -21,7 +21,7 @@ langs = {iso[0]: iso[1] for iso in non_empty_isos} # {'Romanian': 'ro', 'German'
 
 
 models = ["Helsinki-NLP", "Argos", "Google", "t5-base", "t5-small", "t5-large", "Unbabel/Tower-Plus-2B", 
-          "Unbabel/TowerInstruct-Mistral-7B-v0.2", "winninghealth/WiNGPT-Babel-2", "HPLT"]
+          "Unbabel/TowerInstruct-Mistral-7B-v0.2", "winninghealth/WiNGPT-Babel-2", "QUICKMT", "HPLT"]
 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",
@@ -57,6 +57,13 @@ class Translators:
         pipe = pipeline("translation", model=hplt_model, device=self.device)
         translation = pipe(self.input_text)
         return translation[0]['translation_text']
+
+    def quickmt(self)
+        from quickmt import Translator
+        # Auto-detects GPU, set to "cpu" to force CPU inference
+        translation = Translator(f"./quickmt-{self.sl}-{self.tl}/", device="auto")       
+        # Translate - set beam size to 1 for faster speed (but lower quality)
+        return translation({self.input_text}, beam_size=5)
     
     @classmethod
     def download_argos_model(cls, from_code, to_code):
@@ -480,6 +487,9 @@ 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, s_language, t_language, input_text).quickmt()
             
     except Exception as error:
         translated_text = error