Zhibinhong
/

Stable-diffusion-new

Model card Files Files and versions

Zhibinhong commited on Apr 28, 2023

Commit

13b296f

·

1 Parent(s): 890c749

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -20,4 +20,4 @@ class EndpointHandler():
         print(type(image))
         img_data = image.tobytes()
         img_base64 = base64.b64encode(img_data).decode('utf-8')
-        return json.dumps({'image':img_base64})

         print(type(image))
         img_data = image.tobytes()
         img_base64 = base64.b64encode(img_data).decode('utf-8')
+        return {'image':img_base64}