Update src/Translate.py

src/Translate.py

@@ -9,6 +9,22 @@ from transformers import pipeline
 METHOD = "TRANSLATE"
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
+class Translators:
+    def __init__(self, model_name: str, sl: str, tl: str, input_text: str):
+        self.model_name = model_name
+        self.sl, self.tl = sl, tl
+        self.input_text = input_text
+        self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+    def HelsinkiNLP_mulroa(self):
+        try:
+            pipe = pipeline("translation", model=self.model_name, device=self.device)                
+            iso1to3 = {iso[1]: iso[3] for iso in non_empty_isos} # {'ro': 'ron'}
+            iso3tl = iso1to3.get(self.tl) # 'deu', 'ron', 'eng', 'fra'
+            translation = pipe(f'>>{iso3tl}<< {self.input_text}')
+            return translation[0]['translation_text'], f'Translated from {iso1_to_name[self.sl]} to {iso1_to_name[self.tl]} with {self.model_name}.'
+        except Exception as error:
+            return f"Error translating with model: {self.model_name}! Try other available language combination.", error
+
 def paraphraseTranslateMethod(requestValue: str, model: str):
     nltk.download('punkt')
     nltk.download('punkt_tab')