frodos commited on
Commit
54b0e0a
1 Parent(s): 95b5eda

Add result image decode

Browse files
Files changed (1) hide show
  1. handler.py +4 -1
handler.py CHANGED
@@ -25,4 +25,7 @@ class EndpointHandler():
25
  "height": 768,
26
  "width": 768}
27
 
28
- return self._pipe(**params).images[0]
 
 
 
 
25
  "height": 768,
26
  "width": 768}
27
 
28
+ img = self._pipe(**params).images[0]
29
+ res = {"status": 200,
30
+ "image": base64.b64encode(img).decode("utf8")}
31
+ return res