CodeJackR
commited on
Commit
·
06bd1fa
1
Parent(s):
e52ad65
Convert to image segmentation output
Browse files- 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.
|