Update text_to_image.py

text_to_image.py

@@ -8,8 +8,7 @@ if is_accelerate_available():
     from accelerate import PartialState
 
 TEXT_TO_IMAGE_DESCRIPTION = (
-    "This is a tool that creates an image according to a prompt, which is a text description. It takes an input named `prompt` which "
-    "contains the image description and outputs an image."
+    "This is a tool that creates an image according to a prompt, which is a text description."
 )
 
 
@@ -17,7 +16,7 @@ class TextToImageTool(Tool):
     default_checkpoint = "runwayml/stable-diffusion-v1-5"
     description = TEXT_TO_IMAGE_DESCRIPTION
     name = "image_generator"
-    inputs = {"prompt": str}
+    inputs = {"prompt": {"type": str, "description": "contains the image description and outputs an image"}}
     output_type = Image.Image
 
     def __init__(self, device=None, **hub_kwargs) -> None: