debugging

handler.py

@@ -4,7 +4,7 @@ import io
 import tempfile
 from PIL import Image
 from pdf2image import convert_from_path
-
+import logging
 
 def process_image(pdfdata):
     prompt = "detect the accident diagram"
@@ -37,14 +37,15 @@ class EndpointHandler:
         Args:
             path: Path to the model artifacts
         """
-        print("initialized")
+        logging.warning("initialized")
         pass
      
      def __call__(self, data: Any) -> List[List[Dict[str, str]]]:
+         logging.warning("inside __call__")
          inputs = data.pop("inputs", data)
          pdfdata = inputs.pop("filedata", inputs)
          if isinstance(pdfdata, str):
-             print("decoding pdfdata")
+             logging.warning("decoding pdfdata")
              pdfdata = base64.b64decode(pdfdata)
 
          return process_image(pdfdata)