Update handler.py

handler.py

@@ -1,5 +1,5 @@
 from typing import Dict, List, Any
-
+import logging
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
 
@@ -23,6 +23,7 @@ class EndpointHandler():
         conversation = []
         conversation.append({"role": "system", "content": system_prompt})
         conversation.append({"role": "user", "content": message})
+        logging.info(str(conversation))
         input_ids = self.tokenizer.apply_chat_template(conversation, return_tensors="pt")
         input_ids = input_ids.to(self.model.device)