Add result image decode
Browse files- handler.py +4 -1
handler.py
CHANGED
@@ -25,4 +25,7 @@ class EndpointHandler():
|
|
25 |
"height": 768,
|
26 |
"width": 768}
|
27 |
|
28 |
-
|
|
|
|
|
|
|
|
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
|