Vishnu Sarukkai commited on
Commit
d97135c
1 Parent(s): 04b5ed6
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ pipe = StableDiffusionControlNetPipeline.from_pretrained(
16
  pipe.safety_checker = None
17
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
18
  threshold = 250
19
- hed = HEDdetector.from_pretrained('lllyasviel/ControlNet')
20
 
21
  num_images = 3
22
 
 
16
  pipe.safety_checker = None
17
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
18
  threshold = 250
19
+ hed = HEDdetector.from_pretrained('lllyasviel/Annotators')
20
 
21
  num_images = 3
22