Update handler.py

handler.py

@@ -19,10 +19,10 @@ class EndpointHandler:
         
         self.model = YOLO("./best.pt")
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
-        os.getcwd()
         inputs = data.get("inputs")
         isurl = inputs.get("isurl")
         path = inputs.get("path")
+        print(path)
         # path = "http://10.10.2.100/cam-lo.jpg"
         
         model = self.model
@@ -77,14 +77,14 @@ class EndpointHandler:
             
             # cv2.imwrite("result.jpg", clothes)  
         # create base 64 object
-        jpg_as_text = base64.b64encode(buffer).decode("utf-8") # Decode bytes to string")
-        # base64_encoded = base64.b64encode(image_bytes).decode("utf-8")
+        # jpg_as_text = base64.b64encode(buffer).decode("utf-8") # Decode bytes to string")
+        jpg_as_text = base64.b64encode(buffer).decode("utf-8")
         
         # Get the image format
-        image_format = Image.open(io.BytesIO(buffer)).format.lower()
+        # image_format = Image.open(io.BytesIO(buffer)).format.lower()
         
         # Construct the data URI
-        data_uri = f"data:image/{image_format};base64,{jpg_as_text}"
+        # data_uri = f"data:image/{image_format};base64,{jpg_as_text}"
         
         # return data_uri
         print("checkpoint 4")