Update app.py

app.py

@@ -7,27 +7,22 @@ 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` 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": {"type": "str"}  # Must be a dict, not just a type
+        "prompt": {
+            "type": "str",
+            "description": "Text prompt to generate the image"
+        }
     }
+
     output_type = "image"
-    outputs = {
-        "image": "image"
-    }
-    
+
     def __init__(self):
         super().__init__()
         self.client = Client("m-ric/text-to-image")
 
-    def run(self, prompt: str):
-        return self.client.predict(prompt, api_name="/predict")  # Returns PIL.Image
-
+    def run(self, prompt):
+        return self.client.predict(prompt, api_name="/predict")
 
 #%% Utility functions
 def add_label_to_image(image, label):