wolverinn commited on
Commit
b3ff019
·
1 Parent(s): 7a1f53c
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -157,7 +157,7 @@ class EndpointHandler():
157
  args["prompt"] = data["prompt"]
158
  p = StableDiffusionProcessingTxt2Img(sd_model=self.shared.sd_model, **args)
159
  processed = process_images(p)
160
- single_image_b64 = encode_pil_to_base64(processed.images[0])
161
  return {
162
  "img_data": single_image_b64,
163
  }
 
157
  args["prompt"] = data["prompt"]
158
  p = StableDiffusionProcessingTxt2Img(sd_model=self.shared.sd_model, **args)
159
  processed = process_images(p)
160
+ single_image_b64 = encode_pil_to_base64(processed.images[0]).decode('utf-8')
161
  return {
162
  "img_data": single_image_b64,
163
  }