habulaj commited on
Commit
fa3c25f
·
verified ·
1 Parent(s): fdc7882

Update watermark_remover.py

Browse files
Files changed (1) hide show
  1. 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=img_format)
91
 
92
- return out_buffer.getvalue(), img_format
 
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"