Skip to content

Commit 4f84df8

Browse files
committed
Update api.py
1 parent d6c605b commit 4f84df8

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

modules/api/api.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -117,6 +117,8 @@ def mask_decode_to_image(encoding):
117117
# print(f'\nsam/heartbeat: {response.text}\n')
118118
image = encode_pil_to_base64(image)
119119
print(f"mask_decode_to_image image mid 1 {type(image)}")
120+
image = image.decode('utf-8') # Decode bytes to string
121+
print(f"mask_decode_to_image image mid 2 {type(image)}")
120122
dilate_value = 16
121123
response = requests.post('http://0.0.0.0:8080/sam/dilate-mask', json={
122124
"input_image": image,
@@ -126,7 +128,7 @@ def mask_decode_to_image(encoding):
126128
print(f"mask_decode_to_image response 1 masked_image dilate_amount {len(response.dilate_value)}")
127129
if "masked_image" in response:
128130
print(f"mask_decode_to_image response 2 masked_image length {len(response.masked_image)}")
129-
print(f"mask_decode_to_image image mid 2 {type(response.masked_image)}")
131+
print(f"mask_decode_to_image image mid 3 {type(response.masked_image)}")
130132
image = decode_base64_to_image(response.masked_image)
131133
print(f"mask_decode_to_image image after {type(image)}")
132134
print(f'SAM successfully dilated mask by {dilate_value}.')

0 commit comments

Comments
 (0)