sayakpaul HF staff commited on
Commit
b854139
1 Parent(s): 923adf8
Files changed (3) hide show
  1. app.py +2 -2
  2. text_to_image_inpainting.py +1 -1
  3. 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 InpatingTool
3
 
4
- launch_gradio_demo(InpatingTool)
 
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 InpatingTool(Tool):
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.InpatingTool",
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
  }