Update image_transformation.py

image_transformation.py

@@ -21,10 +21,10 @@ class ImageTransformationTool(Tool):
     default_stable_diffusion_checkpoint = "timbrooks/instruct-pix2pix"
     description = IMAGE_TRANSFORMATION_DESCRIPTION
     inputs = {
-        'image': {"type": Image.Image, "description": "the image to transform"},
-        'prompt': {"type": str, "description": "the prompt to use to change the image"}
+        'image': {"type": "image", "description": "the image to transform"},
+        'prompt': {"type": "string", "description": "the prompt to use to change the image"}
     }
-    output_type = Image.Image
+    output_type = "image"
 
     def __init__(self, device=None, controlnet=None, stable_diffusion=None, **hub_kwargs) -> None:
         if not is_accelerate_available():
@@ -51,7 +51,7 @@ class ImageTransformationTool(Tool):
 
         self.is_initialized = True
 
-    def __call__(self, image, prompt):
+    def forward(self, image, prompt):
         if not self.is_initialized:
             self.setup()