Update app.py

app.py

@@ -213,7 +213,7 @@ class Translators:
     def salamandratapipe(self):
         pipe = pipeline("text-generation", model=self.model_name)
         messages = [{"role": "user", "content": 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"][1]["content"]
+        return pipe(messages, max_new_tokens=512, early_stopping=True, num_beams=5)[0]["generated_text"][1]["content"]
     
     def salamandrata(self):
         text = f"Translate the following text from {self.sl} into {self.tl}.\n{self.sl}: {self.input_text} \n{self.tl}:"
@@ -221,7 +221,7 @@ class Translators:
         model = AutoModelForCausalLM.from_pretrained(
             self.model_name,
             device_map="auto",
-            torch_dtype=torch.bfloat16
+            dtype=torch.bfloat16
           )
         message = [{"role": "user", "content": text}]
         # from datetime import datetime