Update image_transformation.py
Browse files- 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}
|