vumichien commited on
Commit
beabc59
·
1 Parent(s): 7c36af7

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +5 -4
utils.py CHANGED
@@ -32,10 +32,11 @@ def read_image_file(file) -> Image.Image:
32
 
33
  def pil_to_base64(img, format="jpeg", encode=False):
34
  if encode:
35
- buffer = BytesIO()
36
- img.save(buffer, format)
37
- buffer.seek(0)
38
- return buffer
 
39
  else:
40
  temp_path = f"temp.{format}"
41
  img.save(temp_path)
 
32
 
33
  def pil_to_base64(img, format="jpeg", encode=False):
34
  if encode:
35
+ bytes_object = BytesIO()
36
+ img.save(bytes_object, format)
37
+ bytes_object.seek(0)
38
+ b64 = base64.b64encode(bytes_object.getvalue()).decode("ascii")
39
+ return b64
40
  else:
41
  temp_path = f"temp.{format}"
42
  img.save(temp_path)