Update app.py

app.py

@@ -93,7 +93,6 @@ class Translators:
             translated_text = f"No Argos model for {self.sl} to {self.tl}. Try other model or languages combination from the available Argos models: {packages_info}."
         except Exception as error:
             translated_text = error
-            print(error)
         return translated_text
     
     def HelsinkiNLP(self):
@@ -114,7 +113,7 @@ class Translators:
                     pipe = pipeline("translation", model=model_name)
                     non_empty_iso = df.slice(1).filter(pl.col("ISO639-1") != "").rows()
                     iso1_dict = {iso[1]: (iso[0], iso[2], iso[3]) for iso in non_empty_iso}
-                    iso3tl = iso1_dict.get(self.tl)[2] # 'deu'
+                    iso3tl = iso1_dict.get(self.tl)[2] # 'deu', 'ron', 'eng', 'fra'
                     translation = pipe(f'>>{iso3tl}<< {self.input_text}')
                     return translation[0]['translation_text'], f'Translated from {self.sl} to {self.tl} with {model_name}.'
                 except Exception as error:
@@ -267,7 +266,7 @@ class Translators:
         inputs = tokenizer(prompt, return_tensors="pt")
         outputs = model.generate(**inputs, max_new_tokens=512)
         output = tokenizer.decode(outputs[0], skip_special_tokens=True)
-        result = output.rsplit(f'{self.tl}:')[-1].strip() if '\n' in output else output.strip()
+        result = output.rsplit(f'{self.tl}:')[-1].strip() if '\n' in output or f'{self.tl}:' in output else output.strip()
         return result
     
     def eurollm_instruct(self):
@@ -355,7 +354,6 @@ def translate_text(input_text: str, s_language: str, t_language: str, model_name
     sl = all_langs[s_language]
     tl = all_langs[t_language]
     message_text = f'Translated from {s_language} to {t_language} with {model_name}'
-    print(message_text)
     try:    
         if model_name.startswith("Helsinki-NLP"):
             translated_text, message_text = Translators(model_name, sl, tl, input_text).HelsinkiNLP()