Upload tool

app.py

@@ -0,0 +1,4 @@
+from transformers import launch_gradio_demo
+from tool import TextToImageTool
+
+launch_gradio_demo(TextToImageTool)

requirements.txt

@@ -0,0 +1,2 @@
+transformers
+huggingface_hub

tool.py

@@ -0,0 +1,16 @@
+from transformers.agents.tools import Tool
+from huggingface_hub import InferenceClient
+
+
+class TextToImageTool(Tool):
+    default_checkpoint = "runwayml/stable-diffusion-v1-5"
+    description = "This is a tool that creates an image according to a prompt, which is a text description."
+    name = "image_generator"
+    inputs = {"prompt": {"type": "text", "description": "the image description"}}
+    output_type = "image"
+    model_sdxl = "stabilityai/stable-diffusion-xl-base-1.0"
+    client = InferenceClient(model_sdxl)
+
+
+    def forward(self, prompt):
+        return self.client.text_to_image(prompt)

tool_config.json

@@ -0,0 +1,7 @@
+{
+  "description": "This is a tool that creates an image according to a prompt, which is a text description.",
+  "inputs": "{'prompt': {'type': 'text', 'description': 'the image description'}}",
+  "name": "image_generator",
+  "output_type": "image",
+  "tool_class": "tool.TextToImageTool"
+}