frodos
/

kandinsky-2-1-img2img

KandinskyPipeline

Inference Endpoints

Model card Files Files and versions Community

frodos commited on Feb 4

Commit

074bf5a

•

1 Parent(s): 1bc1c8f

Define format

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -20,8 +20,8 @@ class EndpointHandler():
         params = {"prompt": inputs.get("prompt", ""),
                   "image": Image.open(BytesIO(base64.b64decode(inputs['image']))),
-                  "strength": inputs.get("strength", 0.3),
-                  "guidance_scale": inputs.get("guidance_scale", 10),
                   "height": 768,
                   "width": 768}

         params = {"prompt": inputs.get("prompt", ""),
                   "image": Image.open(BytesIO(base64.b64decode(inputs['image']))),
+                  "strength": float(inputs.get("strength", 0.3)),
+                  "guidance_scale": float(inputs.get("guidance_scale", 10)),
                   "height": 768,
                   "width": 768}