Spaces:

huggingface-tools
/

text-to-image

Runtime error

lysandre HF Staff commited on May 3, 2023

Commit

9c61b4b

1 Parent(s): c195ce7

IO

Files changed (1) hide show

text_to_image.py CHANGED Viewed

@@ -14,6 +14,8 @@ TEXT_TO_IMAGE_DESCRIPTION = (
 class TextToImageTool(Tool):
     default_checkpoint = "runwayml/stable-diffusion-v1-5"
     description = TEXT_TO_IMAGE_DESCRIPTION
     def __init__(self, device=None, **hub_kwargs) -> None:
         if not is_accelerate_available():

 class TextToImageTool(Tool):
     default_checkpoint = "runwayml/stable-diffusion-v1-5"
     description = TEXT_TO_IMAGE_DESCRIPTION
+    inputs = ['text']
+    outputs = ['image']
     def __init__(self, device=None, **hub_kwargs) -> None:
         if not is_accelerate_available():