m-ric HF staff commited on
Commit
dd47fae
1 Parent(s): 45ce449

Update image_transformation.py

Browse files
Files changed (1) hide show
  1. image_transformation.py +1 -0
image_transformation.py CHANGED
@@ -18,6 +18,7 @@ IMAGE_TRANSFORMATION_DESCRIPTION = (
18
 
19
 
20
  class ImageTransformationTool(Tool):
 
21
  default_stable_diffusion_checkpoint = "timbrooks/instruct-pix2pix"
22
  description = IMAGE_TRANSFORMATION_DESCRIPTION
23
  inputs = {'image': Image.Image, 'prompt': str}
 
18
 
19
 
20
  class ImageTransformationTool(Tool):
21
+ name = "image_transformation"
22
  default_stable_diffusion_checkpoint = "timbrooks/instruct-pix2pix"
23
  description = IMAGE_TRANSFORMATION_DESCRIPTION
24
  inputs = {'image': Image.Image, 'prompt': str}