Update app.py

app.py

@@ -21,6 +21,7 @@ langs.extend(list(all_langs.keys()))
 models = ["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",
+          "Helsinki-NLP/opus-mt-tc-bible-big-roa-deu_eng_fra_por_spa", "Helsinki-NLP/opus-mt-tc-bible-big-deu_eng_fra_por_spa-roa",
           "facebook/nllb-200-distilled-600M", "facebook/nllb-200-distilled-1.3B", "facebook/nllb-200-1.3B", "facebook/nllb-200-3.3B",
           "facebook/mbart-large-50-many-to-many-mmt", "facebook/mbart-large-50-one-to-many-mmt", "facebook/mbart-large-50-many-to-one-mmt",
           "facebook/m2m100_418M", "facebook/m2m100_1.2B",
@@ -102,7 +103,7 @@ class Translators:
             translated_text = error
         return translated_text
 
-    def HelsinkiNLP_mul(self):
+    def HelsinkiNLP_mulroa(self):
         try:
             pipe = pipeline("translation", model=self.model_name, device=self.device)                
             iso_dict = {iso[1]: iso[3] for iso in non_empty_isos}
@@ -141,7 +142,6 @@ class Translators:
         translator = pipeline('translation', model=model, tokenizer=tokenizer, src_lang=self.sl, tgt_lang=self.tl)
         translated_text = translator(text, max_length=512)
         return translated_text[0]['translation_text']
-        # pipe = pipeline("translation", model=)
     
     def smollm(self):
         tokenizer = AutoTokenizer.from_pretrained(self.model_name)
@@ -400,8 +400,8 @@ def translate_text(input_text: str, s_language: str, t_language: str, model_name
     message_text = f'Translated from {s_language} to {t_language} with {model_name}'
     translated_text = None
     try:    
-        if "-mul" in model_name.lower() or "mul-" in model_name.lower():
-            translated_text, message_text = Translators(model_name, sl, tl, input_text).HelsinkiNLP_mul()
+        if "-mul" in model_name.lower() or "mul-" in model_name.lower() or "-roa" in model_name.lower():
+            translated_text, message_text = Translators(model_name, sl, tl, input_text).HelsinkiNLP_mulroa()
         
         elif model_name == "Helsinki-NLP":
             translated_text, message_text = Translators(model_name, sl, tl, input_text).HelsinkiNLP()
@@ -501,7 +501,8 @@ def create_interface():
             swap_btn.click(fn=swap_languages, inputs=[s_language, t_language], outputs=[s_language, t_language], api_name=False, show_api=False)
         # with gr.Row(equal_height=True):
             model_name = gr.Dropdown(choices=models, label=f"Select a model. Default is {DEFAULTS[2]}.", value=DEFAULTS[2], interactive=True, scale=2)
-            # translate_btn = gr.Button("Translate", scale=1)
+            translate_btn = gr.Button(fn=translate_text, inputs=[input_text, s_language, t_language, model_name],
+            outputs=[translated_text, message_text]value="Translate", scale=1)
 
         translated_text = gr.Textbox(label="Translated text:", placeholder="Display field for translation", interactive=False, show_copy_button=True)
         message_text = gr.Textbox(label="Messages:", placeholder="Display field for status and error messages", interactive=False,