Hugging Face's logo

Spaces:
huggingface-tools
/
image-transformation
Build error

App Files Community
6
lysandre HF staff commited on
Commit
5cbbc02
•
1 Parent(s): 6853258

IO

Browse files
Files changed (1) hide show
  1. image_transformation.py +2 -0
image_transformation.py CHANGED
@@ -32,6 +32,8 @@ class ImageTransformationTool(Tool):
32
  default_stable_diffusion_checkpoint = "runwayml/stable-diffusion-v1-5"
33
  default_controlnet_checkpoint = "lllyasviel/sd-controlnet-canny"
34
  description = IMAGE_TRANSFORMATION_DESCRIPTION
 
 
35
 
36
  def __init__(self, device=None, controlnet=None, stable_diffusion=None, **hub_kwargs) -> None:
37
  if not is_accelerate_available():
 
32
  default_stable_diffusion_checkpoint = "runwayml/stable-diffusion-v1-5"
33
  default_controlnet_checkpoint = "lllyasviel/sd-controlnet-canny"
34
  description = IMAGE_TRANSFORMATION_DESCRIPTION
35
+ inputs = ['image', 'text']
36
+ outputs = ['image']
37
 
38
  def __init__(self, device=None, controlnet=None, stable_diffusion=None, **hub_kwargs) -> None:
39
  if not is_accelerate_available():