Update app.py

app.py

@@ -9,9 +9,17 @@ class TextToImageTool(Tool):
     description = "Generate an image from a text prompt using m-ric/text-to-image."
     
     # `inputs` is a dictionary: argument_name -> type
-    inputs = {"prompt": str}
-    output_type = "image" 
-    outputs = {"image": "image"}  # Optional, type can be a string or class
+    #inputs = {"prompt": str}
+    #output_type = "image" 
+    #outputs = {"image": "image"}  # Optional, type can be a string or class
+
+    inputs = {
+        "prompt": {"type": "str"}  # Must be a dict, not just a type
+    }
+    output_type = "image"
+    outputs = {
+        "image": "image"
+    }
     
     def __init__(self):
         super().__init__()