fvonhoven
/

ControlNet-endpoint

stable-diffusion

stable-diffusion-diffusers

endpoints-template

Model card Files Files and versions

fvonhoven commited on Jul 17, 2023

Commit

37b5abc

·

1 Parent(s): c157ba2

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -123,7 +123,7 @@ class EndpointHandler():
         # return first generate PIL image
-        return out.images[0, count]
     # helper to decode input image
     def decode_base64_image(self, image_string):

         # return first generate PIL image
+        return base64.b64encode(out.images[0]).decode('utf-8')
     # helper to decode input image
     def decode_base64_image(self, image_string):