Update app.py

app.py

@@ -210,7 +210,12 @@ class Translators:
             return translation[0]['translation_text'], message
         except Exception as error:
             return f"Error translating with model: {self.model_name}! Try other available language combination or model.", error
-
+    
+    def salamandraTApipe(self):
+        pipe = pipeline("text-generation", model=self.model)
+        messages = [f"Translate the following text from {self.sl} into {self.tl}.\n{self.sl}: {self.input_text} \n{self.tl}:"]
+        return pipe(messages)[0]["generated_text"]
+    
     def salamandraTA(self):
         # from datetime import datetime
         text = f"Translate the following text from {self.sl} into {self.tl}.\n{self.sl}: {self.input_text} \n{self.tl}:"
@@ -266,9 +271,7 @@ class Translators:
    
     def LLaMAX(self):
         pipe = pipeline("text-generation", model="LLaMAX/LLaMAX3-8B")
-        messages = [
-            {"role": "user", "content": f"Translate the following text from {self.sl} to {self.sl}: {self.input_text}"},
-        ]
+        messages = [{"role": "user", "content": f"Translate the following text from {self.sl} to {self.sl}: {self.input_text}"}]
         return pipe(messages)[0]["generated_text"]
     
     def LegoMT(self):
@@ -605,8 +608,8 @@ def translate_text(model_name: str, s_language: str, t_language: str, input_text
         elif model_name == 'Google':
             translated_text = Translators(model_name, sl, tl, input_text).google()
     
-        elif "salamandraTA" in model_name.lower():
-            translated_text = Translators(model_name, sl, tl, input_text).salamandraTA()
+        elif "salamandra" in model_name.lower():
+            translated_text = Translators(model_name, sl, tl, input_text).salamandraTApipe()
 
         elif "m2m" in model_name.lower():
             translated_text = Translators(model_name, sl, tl, input_text).mtom()