Spaces:

m-ric
/

text-to-image

Running

m-ric HF staff commited on May 15, 2024

Commit

bdffc4f

verified ·

1 Parent(s): da3004e

Update tool.py

Files changed (1) hide show

tool.py CHANGED Viewed

@@ -3,7 +3,6 @@ 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 generator prompt. Don't hesitate to add details in the prompt to make the image look better, like 'high-res, photorealistic', etc."}}

 class TextToImageTool(Tool):
     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 generator prompt. Don't hesitate to add details in the prompt to make the image look better, like 'high-res, photorealistic', etc."}}