Convert to image segmentation output

handler.py

@@ -92,7 +92,7 @@ class EndpointHandler():
         
         # 5. Create and return the output PIL Image
         output_img = Image.fromarray(mask_binary)
-        return output_img
+        return [{'score': None, 'label': 'everything', 'mask': output_img}]
 
 def main():
     # This main function shows how a client would call the endpoint locally.