asoderznik
/

sdx4-upscaler

StableDiffusionUpscalePipeline

stable-diffusion

Inference Endpoints

Model card Files Files and versions Community

asoderznik commited on Aug 29, 2023

Commit

d188461

•

1 Parent(s): ba24c0a

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -54,5 +54,5 @@ class EndpointHandler():
         img_str = base64.b64encode(buffered.getvalue())
         # postprocess the prediction
-        return {"image": img_str}
         #return {"image": "test"}

         img_str = base64.b64encode(buffered.getvalue())
         # postprocess the prediction
+        return {"image": img_str.decode()}
         #return {"image": "test"}