Zhibinhong
/

Stable-diffusion-new

Inference Endpoints

Model card Files Files and versions Community

Zhibinhong commited on May 3, 2023

Commit

f17ad12

•

1 Parent(s): 8941627

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -20,5 +20,5 @@ class EndpointHandler():
         img = self.pipe(text).images[0]
         img.save("./1.png")
         with open('./1.png','rb') as img_file:
-            encoded_string = base64.b64encode(img_file.read())
         return {'image':encoded_string}

         img = self.pipe(text).images[0]
         img.save("./1.png")
         with open('./1.png','rb') as img_file:
+            encoded_string = base64.b64encode(img_file.read()).decode('utf-8')
         return {'image':encoded_string}