Update handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
@@ -40,7 +40,7 @@ class EndpointHandler:
|
|
40 |
|
41 |
# encode image as base 64
|
42 |
buffered = BytesIO()
|
43 |
-
image.save(buffered, format="
|
44 |
img_str = b64encode(buffered.getvalue())
|
45 |
|
46 |
# postprocess the prediction
|
|
|
40 |
|
41 |
# encode image as base 64
|
42 |
buffered = BytesIO()
|
43 |
+
image.save(buffered, format="PNG")
|
44 |
img_str = b64encode(buffered.getvalue())
|
45 |
|
46 |
# postprocess the prediction
|