Update watermark_remover.py
Browse files- watermark_remover.py +2 -2
watermark_remover.py
CHANGED
|
@@ -87,6 +87,6 @@ def remove_watermark(image_bytes: bytes) -> tuple[bytes, str]:
|
|
| 87 |
# Create output image
|
| 88 |
out_img = Image.fromarray(data.astype(np.uint8), mode=img.mode)
|
| 89 |
out_buffer = io.BytesIO()
|
| 90 |
-
out_img.save(out_buffer, format=
|
| 91 |
|
| 92 |
-
return out_buffer.getvalue(),
|
|
|
|
| 87 |
# Create output image
|
| 88 |
out_img = Image.fromarray(data.astype(np.uint8), mode=img.mode)
|
| 89 |
out_buffer = io.BytesIO()
|
| 90 |
+
out_img.save(out_buffer, format="PNG")
|
| 91 |
|
| 92 |
+
return out_buffer.getvalue(), "PNG"
|