Spaces:

huggingface-tools
/

text-to-image

Runtime error

m-ric HF staff commited on 28 days ago

Commit

e8e18da

•

1 Parent(s): 5f5b2e1

Update text_to_image.py

Files changed (1) hide show

text_to_image.py CHANGED Viewed

@@ -15,7 +15,7 @@ class TextToImageTool(Tool):
     default_checkpoint = "runwayml/stable-diffusion-v1-5"
     description = TEXT_TO_IMAGE_DESCRIPTION
     name = "image_generator"
-    inputs = {"prompt": {"type": "text", "description": "the image description"}}
     output_type = "image"
     def __init__(self, device=None, **hub_kwargs) -> None:

     default_checkpoint = "runwayml/stable-diffusion-v1-5"
     description = TEXT_TO_IMAGE_DESCRIPTION
     name = "image_generator"
+    inputs = {"prompt": {"type": "string", "description": "the image description"}}
     output_type = "image"
     def __init__(self, device=None, **hub_kwargs) -> None: