Update app.py

app.py

@@ -7,10 +7,10 @@ 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": {
-            "type": "string",
+            "type": "str",
             "description": "Text prompt to generate the image"
         }
     }
@@ -21,7 +21,7 @@ class TextToImageTool(Tool):
         super().__init__()
         self.client = Client("m-ric/text-to-image")
 
-    def run(self, prompt):
+    def run(self, prompt):  # Must explicitly match 'inputs' keys
         return self.client.predict(prompt, api_name="/predict")
 
 #%% Utility functions