Spaces:
Runtime error
Runtime error
add: inputs and outputs.
Browse files
text_to_image_inpainting.py
CHANGED
@@ -18,7 +18,7 @@ INPAINTING_DESCRIPTION = (
|
|
18 |
class InpatingTool(Tool):
|
19 |
default_checkpoint = "stabilityai/stable-diffusion-2-inpainting"
|
20 |
description = INPAINTING_DESCRIPTION
|
21 |
-
inputs = ['text']
|
22 |
outputs = ['image']
|
23 |
|
24 |
def __init__(self, device=None, **hub_kwargs) -> None:
|
|
|
18 |
class InpatingTool(Tool):
|
19 |
default_checkpoint = "stabilityai/stable-diffusion-2-inpainting"
|
20 |
description = INPAINTING_DESCRIPTION
|
21 |
+
inputs = ['image', 'image', 'text']
|
22 |
outputs = ['image']
|
23 |
|
24 |
def __init__(self, device=None, **hub_kwargs) -> None:
|