Update app.py

app.py

@@ -10,6 +10,7 @@ class TextToImageTool(Tool):
     
     # `inputs` is a dictionary: argument_name -> type
     inputs = {"prompt": str}
+    output_type = "image" 
     outputs = {"image": "image"}  # Optional, type can be a string or class
     
     def __init__(self):