CodeJackR commited on
Commit
06bd1fa
·
1 Parent(s): e52ad65

Convert to image segmentation output

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -92,7 +92,7 @@ class EndpointHandler():
92
 
93
  # 5. Create and return the output PIL Image
94
  output_img = Image.fromarray(mask_binary)
95
- return output_img
96
 
97
  def main():
98
  # This main function shows how a client would call the endpoint locally.
 
92
 
93
  # 5. Create and return the output PIL Image
94
  output_img = Image.fromarray(mask_binary)
95
+ return [{'score': None, 'label': 'everything', 'mask': output_img}]
96
 
97
  def main():
98
  # This main function shows how a client would call the endpoint locally.