Update handler.py

handler.py

@@ -4,16 +4,17 @@ import requests
 from io import BytesIO
 from PIL import Image
 import time
+import logging
 import os
 
 
 class EndpointHandler():
     def __init__(self, path=""):
         # load the model
-        print(f'value of path is : {path}')
+        logging.info(f'value of path is : {path}')
         current_directory = os.getcwd()
-        print(f'current dir: {current_directory}')
-        print(f'all files: {os.listdir(current_directory)}')
+        logging.info(f'current dir: {current_directory}')
+        logging.info(f'all files: {os.listdir(current_directory)}')
         # model_path = "yolov8m_detect_usdl.pt"
         # self.model = YOLO(model_path)
         # print('Yolo model loaded successfully')
@@ -21,7 +22,7 @@ class EndpointHandler():
 
     def __call__(self, data: Dict) -> Dict:
 
-        print(f'Value of data is : {data}')
+        logging.info(f'Value of data is : {data}')
 
         # image_url = data.get('image_url')
         # print(f'Image url is : {image_url}')
@@ -33,4 +34,4 @@ class EndpointHandler():
         # results = self.model(pil_image)
         # print(f'TIME Model inference: {time.time() - t1}')
         # postprocess the prediction -> results[0].boxes.data.tolist()
-        return {"bbox": 'data'}
+        return {"bbox": data}