Update handler.py

handler.py

@@ -7,7 +7,7 @@ class EndpointHandler():
         self.pipe = pipeline("image-segmentation", model=path, trust_remote_code=True)
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         # Extract the image path from the input data
-        image_path = data.pop("image_path",data)
+        image_path = data.pop("inputs",data)
         
         # Perform image segmentation
         pillow_mask = self.pipe(image_path, return_mask=True)  # outputs a pillow mask