Change inference flag, use byte instead of base64

README.md

@@ -5,7 +5,7 @@ tags:
 - stable-diffusion-diffusers
 - text-to-image
 - endpoints-template
-inference: false
+inference: true
 duplicated_from: philschmid/stable-diffusion-v1-4-endpoints
 ---
 # Fork of [CompVis/stable-diffusion-v1-4](https://huggingface.co/CompVis/stable-diffusion-v1-4)

handler.py

@@ -42,7 +42,6 @@ class EndpointHandler:
         # encode image as base 64
         buffered = BytesIO()
         image.save(buffered, format="png")
-        img_str = base64.b64encode(buffered.getvalue())
 
         # post process the prediction
-        return {"image": img_str.decode()}
+        return {"image": buffered.getvalue()}