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}
|