Update app.py

app.py

@@ -99,11 +99,11 @@ class Translators:
             pipe = pipeline("translation", model=hplt_model, device=self.device)
             translation = pipe(self.input_text)
             translated_text = translation[0]['translation_text']
-            message = f'Translated from {iso1_to_name[self.sl]} to {iso1_to_name[self.tl]} with {hplt_model}.'
+            message_text = f'Translated from {iso1_to_name[self.sl]} to {iso1_to_name[self.tl]} with {hplt_model}.'
         else:
             translated_text = f'HPLT model from {iso1_to_name[self.sl]} to {iso1_to_name[self.tl]} not available!'
-            message = f"Available models: {', '.join(hplt_models)}"
-        return translated_text, message
+            message_text = f"Available models: {', '.join(hplt_models)}"
+        return translated_text, message_text
 
     @staticmethod
     def download_argos_model(from_code, to_code):
@@ -539,9 +539,9 @@ def translate_text(input_text: str, s_language: str, t_language: str, model_name
 
         elif "HPLT" in model_name:
             if model_name == "HPLT-OPUS":
-                translated_text, message = Translators(model_name, sl, tl, input_text).hplt(opus = True)
+                translated_text, message_text = Translators(model_name, sl, tl, input_text).hplt(opus = True)
             else:
-                translated_text, message = Translators(model_name, sl, tl, input_text).hplt()
+                translated_text, message_text = Translators(model_name, sl, tl, input_text).hplt()
         
         elif model_name == "Bergamot":
             translated_text, message_text = Translators(model_name, s_language, t_language, input_text).bergamot()
@@ -567,7 +567,9 @@ def get_info(model_name: str, sl: str = None, tl: str = None):
     elif model_name == "Argos":
         return httpx.get(f'https://huggingface.co/TiberiuCristianLeon/Argostranslate/raw/main/README.md').text
     elif "HPLT" in model_name:
-        return '[HPLT Uni direction translation models](https://huggingface.co/collections/HPLT/hplt-12-uni-direction-translation-models)'        
+        return '[HPLT Uni direction translation models](https://huggingface.co/collections/HPLT/hplt-12-uni-direction-translation-models)'
+    elif "QUICKMT" in model_name:
+        return '[QUICKMT](https://huggingface.co/quickmt)'  
     elif model_name == "Google":
         return "Google Translate Online"
     else: