Spaces:

mikonvergence
/

mask-and-sketch

Sleeping

mikonvergence commited on Jul 7, 2023

Commit

fe198cf

•

1 Parent(s): 807dfed

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -26,7 +26,7 @@ from src.pipeline_stable_diffusion_controlnet_inpaint import *
 from diffusers.utils import load_image
 from controlnet_aux import HEDdetector
-hed = HEDdetector.from_pretrained('lllyasviel/ControlNet')
 controlnet = ControlNetModel.from_pretrained(
     "fusing/stable-diffusion-v1-5-controlnet-scribble", torch_dtype=torch.float16

 from diffusers.utils import load_image
 from controlnet_aux import HEDdetector
+hed = HEDdetector.from_pretrained('lllyasviel/Annotators')
 controlnet = ControlNetModel.from_pretrained(
     "fusing/stable-diffusion-v1-5-controlnet-scribble", torch_dtype=torch.float16