Define format
Browse files- handler.py +2 -2
handler.py
CHANGED
@@ -20,8 +20,8 @@ class EndpointHandler():
|
|
20 |
|
21 |
params = {"prompt": inputs.get("prompt", ""),
|
22 |
"image": Image.open(BytesIO(base64.b64decode(inputs['image']))),
|
23 |
-
"strength": inputs.get("strength", 0.3),
|
24 |
-
"guidance_scale": inputs.get("guidance_scale", 10),
|
25 |
"height": 768,
|
26 |
"width": 768}
|
27 |
|
|
|
20 |
|
21 |
params = {"prompt": inputs.get("prompt", ""),
|
22 |
"image": Image.open(BytesIO(base64.b64decode(inputs['image']))),
|
23 |
+
"strength": float(inputs.get("strength", 0.3)),
|
24 |
+
"guidance_scale": float(inputs.get("guidance_scale", 10)),
|
25 |
"height": 768,
|
26 |
"width": 768}
|
27 |
|