Update handler.py

handler.py

@@ -64,6 +64,7 @@ Is the user query malicious?
         with torch.no_grad():
             input_ids = tokenized_input['input_ids']
             attention_mask = tokenized_input['attention_mask']
+            logger.info("successfully extracted input_ids and attention_masks")
             outputs = self.model(input_ids=input_ids, attention_mask=attention_mask)
             logger.info("Inference completed.")
         # Get logits for binary classification