Update src/Translate.py

src/Translate.py

@@ -20,7 +20,7 @@ class Translators:
         self.sl, self.tl = sl, tl
         self.input_text = input_text
         self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-    message = f'Translated from {iso1_to_name[self.sl]} to {iso1_to_name[self.tl]} with {self.model_name}.'
+        self.message = f'Translated from {iso1_to_name[self.sl]} to {iso1_to_name[self.tl]} with {self.model_name}.'
     def HelsinkiNLP_mulroa(self):
         try:
             pipe = pipeline("translation", model=self.model_name, device=self.device)                
@@ -32,10 +32,10 @@ class Translators:
             return f"Error translating with model: {self.model_name}! Try other available language combination.", error
     def text2textgenerationpipe(self):
         translation  = pipeline('text2text-generation', model = self.model_name)
-        return translation(text)[0]['generated_text'], message
+        return translation(text)[0]['generated_text'], self.message
     def translationpipe(self):
         translation  = pipeline('translation', model = self.model_name)
-        return translation(text)[0]['translation_text'], message
+        return translation(text)[0]['translation_text'], self.message
 
 def paraphraseTranslateMethod(requestValue: str, model: str):
     nltk.download('punkt')