Update app.py

app.py

@@ -7,14 +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
 
     def __init__(self):
         super().__init__()
-        self.client = Client("m-ric/text-to-image")  # Calls HF Space
+        self.client = Client("m-ric/text-to-image")
 
     def run(self, prompt: str):
         image = self.client.predict(prompt, api_name="/predict")
-        return image  # This is a PIL image
+        return image  # PIL.Image
+
 
 #%% Utility functions
 def add_label_to_image(image, label):