simulationcartridge
/

ppl

stable-diffusion

template:sd-lora

Model card Files Files and versions

Mirza Učanbarlić commited on Nov 12, 2023

Commit

b074d05

·

1 Parent(s): c058aa4

fix img_str

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -37,7 +37,7 @@ class EndpointHandler():
         # encode image as base 64
         buffered = BytesIO()
         image.save(buffered, format="JPEG")
-        img_str = base64.b64encode(buffered.getvalue())
         # postprocess the prediction
         return {"image": img_str}

         # encode image as base 64
         buffered = BytesIO()
         image.save(buffered, format="JPEG")
+        img_str = base64.b64encode(buffered.getvalue()).decode('utf-8')
         # postprocess the prediction
         return {"image": img_str}