Update app.py

app.py

@@ -7,16 +7,17 @@ from gradio_client import Client
 class TextToImageTool(Tool):
     name = "text_to_image"
     description = "Generate an image from a text prompt using m-ric/text-to-image."
-    inputs = ["prompt"]          # Required by smolagents
-    outputs = ["image"]          # Not used directly but required
-
+    
+    # `inputs` is a dictionary: argument_name -> type
+    inputs = {"prompt": str}
+    outputs = {"image": "image"}  # Optional, type can be a string or class
+    
     def __init__(self):
         super().__init__()
         self.client = Client("m-ric/text-to-image")
 
     def run(self, prompt: str):
-        image = self.client.predict(prompt, api_name="/predict")
-        return image  # PIL.Image
+        return self.client.predict(prompt, api_name="/predict")  # Returns PIL.Image
 
 
 #%% Utility functions