Update app.py

app.py

@@ -7,6 +7,7 @@ import languagecodes
 favourite_langs = {"German": "de", "Romanian": "ro", "English": "en", "-----": "-----"}
 all_langs = languagecodes.iso_languages
 
+# Language options as list, add favourite languages first
 options = list(favourite_langs.keys())
 options.extend(list(all_langs.keys()))
 models = ["Helsinki-NLP", "t5-base", "t5-small", "t5-large", "facebook/nllb-200-distilled-600M", "facebook/nllb-200-distilled-1.3B", "facebook/mbart-large-50-many-to-many-mmt"]
@@ -23,7 +24,6 @@ def model_to_cuda(model):
 def translate_text(input_text, sselected_language, tselected_language, model_name):
     sl = all_langs[sselected_language]
     tl = all_langs[tselected_language]
-    message_text = f'Translated from {sselected_language} to {tselected_language} with {model_name}'
     if model_name == "Helsinki-NLP":
         try:
             model_name = f"Helsinki-NLP/opus-mt-{sl}-{tl}"
@@ -70,6 +70,7 @@ def translate_text(input_text, sselected_language, tselected_language, model_nam
     output_ids = model.generate(input_ids, max_length=512)
     translated_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
     
+    message_text = f'Translated from {sselected_language} to {tselected_language} with {model_name}'
     print(f'Translating from {sselected_language} to {tselected_language} with {model_name}:', f'{input_text} =  {translated_text}', sep='\n')
     return translated_text, message_text