Update handler.py

handler.py

@@ -23,15 +23,16 @@ class EndpointHandler():
     def __call__(self, data: Dict) -> Dict:
 
         # logging.info(f'Value of data is : {data}')
-
+        logger.info(f'data: {data}')
+        logger.info(f'type of data: {type(data)}')
         image_url = data.get('image_url')
-        print(f'Image url is : {image_url}')
+        logger.info(f'Image url is : {image_url}')
         response = requests.get(image_url)
         pil_image = Image.open(BytesIO(response.content))
         
-        print('Model inference started....')
+        logger.info('Model inference started....')
         t1 = time.time()
         results = self.model(pil_image)
-        print(f'TIME Model inference: {time.time() - t1}')
+        logger.info(f'TIME Model inference: {time.time() - t1}')
         # postprocess the prediction -> results[0].boxes.data.tolist()
         return {"bbox": results[0].boxes.data.tolist()}