Update tool.py

tool.py

@@ -5,7 +5,7 @@ from huggingface_hub import InferenceClient
 
 
 class TextToImageTool(Tool):
-    description = "This tool creates an image according to a prompt, which is a text description."
+    description = "This tool creates an image according to a prompt, which is a text description. When using this tool in a code snippet, the last line of the snippet should be the the tool call (so the image is shown on screen)."
     name = "image_generator"
     inputs = {"prompt": {"type": "string", "description": "The image generator prompt. Don't hesitate to add details in the prompt to make the image look better, like 'high-res, photorealistic', etc."},
              "save_path": {"type": "string", "description": "A file path in `/tmp` to save the image to. The file path extenstion should be .png", "nullable": True}