Update app.py

app.py

@@ -212,16 +212,16 @@ class Translators:
             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)
+        pipe = pipeline("text-generation", model=self.model_name)
         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}:"
-        tokenizer = AutoTokenizer.from_pretrained(self.model)        
+        tokenizer = AutoTokenizer.from_pretrained(self.model_name)        
         model = AutoModelForCausalLM.from_pretrained(
-            self.model,
+            self.model_name,
             device_map="auto",
             torch_dtype=torch.bfloat16
           )