Update image_transformation.py

image_transformation.py

@@ -11,8 +11,7 @@ if is_accelerate_available():
     from accelerate import PartialState
 
 IMAGE_TRANSFORMATION_DESCRIPTION = (
-    "This is a tool that transforms an image according to a prompt. It takes two inputs: `image`, which should be "
-    "the image to transform, and `prompt`, which should be the prompt to use to change it. It returns the "
+    "This is a tool that transforms an image according to a prompt and returns the "
     "modified image."
 )
 
@@ -21,7 +20,10 @@ class ImageTransformationTool(Tool):
     name = "image_transformation"
     default_stable_diffusion_checkpoint = "timbrooks/instruct-pix2pix"
     description = IMAGE_TRANSFORMATION_DESCRIPTION
-    inputs = {'image': Image.Image, 'prompt': str}
+    inputs = {
+        'image': {"type": Image.Image, "description": "the image to transform"},
+        'prompt': {"type": str, "description": "the prompt to use to change the image"}
+    }
     output_type = Image.Image
 
     def __init__(self, device=None, controlnet=None, stable_diffusion=None, **hub_kwargs) -> None: