blob: be1dbda3d3ebe8e480abf6ca1df088ae892f0757 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
from __future__ import annotations
from ..schema import GenerationResult, InpaintImageRequest
from ..service import ImageGenerationService
async def inpaint_image(
service: ImageGenerationService,
positive_prompt: str,
input_image: str,
mask_image: str,
negative_prompt: str | None = None,
size: str | None = None,
) -> GenerationResult:
request = InpaintImageRequest.create(
positive_prompt,
input_image,
mask_image,
negative_prompt,
size,
default_size=service.config.generation.default_size,
)
return await service.inpaint_image(request)
|
