merve HF staff commited on
Commit
f03a9f0
1 Parent(s): 9471efd

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -52,7 +52,7 @@ class PreTrainedPipeline():
52
  with io.BytesIO() as out:
53
  img.save(out, format="PNG")
54
  png_string = out.getvalue()
55
- mask = base64.b64encode(png_string).decode("utf-8")
56
 
57
  mask_codes[f"mask_{cls}"] = mask
58
 
 
52
  with io.BytesIO() as out:
53
  img.save(out, format="PNG")
54
  png_string = out.getvalue()
55
+ mask = base64.b64encode(png_string)
56
 
57
  mask_codes[f"mask_{cls}"] = mask
58