Spaces:
Runtime error
Runtime error
fix typo.
Browse files- app.py +2 -2
- text_to_image_inpainting.py +1 -1
- tool_config.json +1 -1
app.py
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
from transformers.tools.base import launch_gradio_demo
|
2 |
-
from text_to_image_inpainting import
|
3 |
|
4 |
-
launch_gradio_demo(
|
|
|
1 |
from transformers.tools.base import launch_gradio_demo
|
2 |
+
from text_to_image_inpainting import InpaintingTool
|
3 |
|
4 |
+
launch_gradio_demo(InpaintingTool)
|
text_to_image_inpainting.py
CHANGED
@@ -15,7 +15,7 @@ INPAINTING_DESCRIPTION = (
|
|
15 |
)
|
16 |
|
17 |
|
18 |
-
class
|
19 |
default_checkpoint = "stabilityai/stable-diffusion-2-inpainting"
|
20 |
description = INPAINTING_DESCRIPTION
|
21 |
inputs = ['image', 'image', 'text']
|
|
|
15 |
)
|
16 |
|
17 |
|
18 |
+
class InpaintingTool(Tool):
|
19 |
default_checkpoint = "stabilityai/stable-diffusion-2-inpainting"
|
20 |
description = INPAINTING_DESCRIPTION
|
21 |
inputs = ['image', 'image', 'text']
|
tool_config.json
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
{
|
2 |
-
"tool_class": "text_to_image_inpainting.
|
3 |
"description": "This is a tool that inpaints some parts of an image StableDiffusionInpaintPipeline according to a prompt. It takes three inputs: `image`, which should be the original image which will be inpainted, `mask_image`, which should be used to determine which parts of the original image (stored in the `image` variable) should be inpainted, and `prompt`, which should be the prompt to use to guide the inpainting process. It returns the inpainted image.",
|
4 |
"name": "image_inpainter"
|
5 |
}
|
|
|
1 |
{
|
2 |
+
"tool_class": "text_to_image_inpainting.InpaintingTool",
|
3 |
"description": "This is a tool that inpaints some parts of an image StableDiffusionInpaintPipeline according to a prompt. It takes three inputs: `image`, which should be the original image which will be inpainted, `mask_image`, which should be used to determine which parts of the original image (stored in the `image` variable) should be inpainted, and `prompt`, which should be the prompt to use to guide the inpainting process. It returns the inpainted image.",
|
4 |
"name": "image_inpainter"
|
5 |
}
|