Image masking support

2022-08-26 18:56:34 +05:30
parent 252900356f
commit 3de2383505
2 changed files with 66 additions and 4 deletions
--- a/main.py
+++ b/main.py
@ -13,6 +13,7 @@ app = FastAPI()
 class ImageRequest(BaseModel):
    prompt: str
    init_image: str = None # base64
+    mask: str = None # base64
    num_outputs: str = "1"
    num_inference_steps: str = "50"
    guidance_scale: str = "7.5"
@ -51,6 +52,9 @@ async def image(req : ImageRequest):
        data['input']['init_image'] = req.init_image
        data['input']['prompt_strength'] = req.prompt_strength

+        if req.mask is not None:
+            data['input']['mask'] = req.mask
+
    if req.seed == "-1":
        del data['input']['seed']