serhatderya
/

v1_1_scribble

Model card Files Files and versions

serhatderya commited on Jun 22, 2023

Commit

946d8b7

·

1 Parent(s): c6d41c8

Update handler.py

Files changed (1) hide show

handler.py +4 -2

handler.py CHANGED Viewed

@@ -50,6 +50,8 @@ class EndpointHandler:
         image = pipe(prompt, num_inference_steps=30, generator=generator, image=control_image).images[0]
         # postprocess the prediction
-        res_base64 = base64.b64encode(image)
-        return [{"result": res_base64}]

         image = pipe(prompt, num_inference_steps=30, generator=generator, image=control_image).images[0]
         # postprocess the prediction
+        buffered = BytesIO()
+        image.save(buffered, format="JPEG")
+        img_str = base64.b64encode(buffered.getvalue())
+        return {"result": img_str.decode()}