wolverinn
commited on
Commit
·
b3ff019
1
Parent(s):
7a1f53c
success
Browse files- 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 |
}
|