Update app.py

app.py

@@ -36,13 +36,14 @@ models = ["Helsinki-NLP", "QUICKMT", "Argos", "Lego-MT/Lego-MT", "HPLT", "HPLT-O
           "Helsinki-NLP/opus-mt-tc-bible-big-roa-en",
           "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", "alirezamsh/small100",
+          "facebook/m2m100_418M", "facebook/m2m100_1.2B",
+          "alirezamsh/small100", "naist-nlp/mitre_466m", "naist-nlp/mitre_913m",
           "bigscience/mt0-small", "bigscience/mt0-base", "bigscience/mt0-large", "bigscience/mt0-xl",
           "bigscience/bloomz-560m", "bigscience/bloomz-1b1", "bigscience/bloomz-1b7", "bigscience/bloomz-3b",
           "t5-small", "t5-base", "t5-large",
           "google/flan-t5-small", "google/flan-t5-base", "google/flan-t5-large", "google/flan-t5-xl",
           "google/madlad400-3b-mt", "Heng666/madlad400-3b-mt-ct2", "Heng666/madlad400-3b-mt-ct2-int8", "Heng666/madlad400-7b-mt-ct2-int8",
-          "naist-nlp/mitre_466m", "naist-nlp/mitre_913m",
+          "yanolja/YanoljaNEXT-Rosetta-4B-2511"
           "BSC-LT/salamandraTA-2b-instruct", "BSC-LT/salamandraTA-7b-instruct",
           "utter-project/EuroLLM-1.7B", "utter-project/EuroLLM-1.7B-Instruct",
          "Unbabel/Tower-Plus-2B", "Unbabel/TowerInstruct-7B-v0.2", "Unbabel/TowerInstruct-Mistral-7B-v0.2",
@@ -64,7 +65,7 @@ class Translators:
         url = os.environ['GCLIENT'] + f'sl={self.sl}&tl={self.tl}&q={self.input_text}'
         response = httpx.get(url)
         return response.json()[0][0][0].strip()
-
+   
     def mitre(self):
         from transformers import AutoModel, AutoTokenizer
         tokenizer = AutoTokenizer.from_pretrained(self.model_name, trust_remote_code=True, use_fast=False)
@@ -236,7 +237,7 @@ class Translators:
     def simplepipe(self):
         try:
             pipe = pipeline("translation", model=self.model_name, device=self.device)                
-            translation = pipe(self.input_text)
+            translation = pipe(self.input_text, , src_lang=self.sl, tgt_lang=self.tl)
             message = f'Translated from {iso1toall[self.sl][0]} to {iso1toall[self.tl][0]} with {self.model_name}.'
             return translation[0]['translation_text'], message
         except Exception as error:
@@ -630,6 +631,9 @@ def translate_text(model_name: str, s_language: str, t_language: str, input_text
         elif "salamandra" in model_name.lower():
             translated_text = Translators(model_name, s_language, t_language, input_text).salamandratapipe()
         
+        elif "Rosetta" in model_name.lower():
+            translated_text = Translators(model_name, s_language, t_language, input_text).simplepipe()
+        
         elif "small100" in model_name.lower():
             translated_text = Translators(model_name, sl, tl, input_text).smallonehundred()