Update handler.py

handler.py

@@ -13,7 +13,7 @@ class EndpointHandler():
         
         # Perform image segmentation
         pillow_mask = self.pipe(image_path, return_mask=True)  # outputs a pillow mask
-        pillow_image = self.pipe(image_path, return_mask=True)  # outputs the segmented image
+        pillow_image = self.pipe(image_path)  # outputs the segmented image
         
         # Return the result as a list of dictionaries
         return [{"image": pillow_image, "mask": pillow_mask}]