Update handler.py

handler.py

@@ -10,8 +10,8 @@ class EndpointHandler():
         image_path = data.pop("inputs",data)
         
         # Perform image segmentation
-        pillow_mask = self.pipe(inputs, return_mask=True)  # outputs a pillow mask
-        pillow_image = self.pipe(inputs)  # outputs the segmented image
+        pillow_mask = self.pipe(image_path, return_mask=True)  # outputs a pillow mask
+        pillow_image = self.pipe(image_path)  # outputs the segmented image
         
         # Save the segmented image at the root folder
         output_image_path = "segmented_image.png"