Update handler2.py

handler2.py

@@ -65,7 +65,14 @@ class EndpointHandler:
         
         # Define the policy prompt for safety evaluation
         self.policy_prompt = self._get_policy_prompt()
- 
+
+
+        try:
+            template_tokenizer = self.tokenizer
+            if hasattr(template_tokenizer, "chat_template") and template_tokenizer.chat_template:
+                self.tokenizer.chat_template = template_tokenizer.chat_template
+                print(f"Successfully imported chat template from {self.chat_template_id}")\
+        
 
     def _get_policy_prompt(self):
         return """Check if there is unsafe content in the user or agent messages in the