Update app.py

app.py

@@ -62,21 +62,23 @@ class Translators:
         from quickmt import Translator
         from quickmt.hub import hf_download, hf_list
         from pathlib import Path
-        input_model = f"quickmt-{self.sl}-{self.tl}"
-        model_path = Path("/code/models") / input_model
+        model_name = f"quickmt-{self.sl}-{self.tl}"
+        model_path = Path("/code/models") / model_name
         choices = [i.split("/")[1] for i in hf_list()]
         print(choices)
         if not model_path.exists():
             hf_download(
-            model_name = f"quickmt/{input_model}",
-            output_dir=Path("/code/models") / input_model,
+            model_name = f"quickmt/{model_name}",
+            output_dir=Path("/code/models") / model_name,
         )
-        print(f"Loading model {input_model}")
+        print(f"Loading model {model_name}")
         # Auto-detects GPU, set to "cpu" to force CPU inference
-        translation = Translator(str(model_path), device="auto", inter_threads=2)       
+        translator = Translator(str(model_path), device="auto", inter_threads=2)       
         # translation = Translator(f"./quickmt-{self.sl}-{self.tl}/", device="auto", inter_threads=2)       
         # Translate - set beam size to 1 for faster speed (but lower quality)
-        return translation({self.input_text}, beam_size=5)
+        translation = translator({self.input_text}, beam_size=5)
+        message = f'Translated from {iso1_to_name[self.sl]} to {iso1_to_name[self.tl]} with {model_name}.'
+        return translation, message
     
     @classmethod
     def download_argos_model(cls, from_code, to_code):