Update app.py

app.py

@@ -224,24 +224,24 @@ class Translators:
             dtype=torch.bfloat16
           )
         message = [{"role": "user", "content": text}]
-        # from datetime import datetime
-        # date_string = datetime.today().strftime('%Y-%m-%d')
+        from datetime import datetime
+        date_string = datetime.today().strftime('%Y-%m-%d')
         prompt = tokenizer.apply_chat_template(
             message,
             tokenize=False,
-            add_generation_prompt=True
+            add_generation_prompt=True,
+            date_string=date_string
         )
-        # date_string=date_string
-        inputs = tokenizer(prompt, add_special_tokens=False, return_tensors="pt")
+        inputs = tokenizer.encode(prompt, add_special_tokens=False, return_tensors="pt")
         attention_mask = inputs["attention_mask"]        
         input_length = inputs.shape[1]
-        outputs = model.generate(inputs['input_ids'].to(model.device),
+        outputs = model.generate(input_ids=inputs.to(model.device),
                                 max_new_tokens=512,
                                 early_stopping=True,
                                 num_beams=5,
                                 attention_mask=attention_mask,
-                                pad_token_id=tokenizer.eos_token_id)
-        # eos_token_id=tokenizer.eos_token_id
+                                pad_token_id=tokenizer.eos_token_id,
+                                eos_token_id=tokenizer.eos_token_id)
         return tokenizer.decode(outputs[0, input_length:], skip_special_tokens=True)
     
     def HelsinkiNLP_mulroa(self):