Spaces:

ysmao
/

Layout-Control

Running on Zero

ysmao commited on May 30

Commit

7d60aec

•

1 Parent(s): b3e1c9e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ from annotator.midas import DepthDetector
 from annotator.dsine_local import NormalDetector
 from annotator.upernet import SegmDetector
-controlnet_checkpoint = "kujiale-ai/controlnet"
 # Initialize pipeline
 controlnet = ControlNetModel.from_pretrained(
     controlnet_checkpoint,

 from annotator.dsine_local import NormalDetector
 from annotator.upernet import SegmDetector
+controlnet_checkpoint = "kujiale-ai/controlnet-layout"
 # Initialize pipeline
 controlnet = ControlNetModel.from_pretrained(
     controlnet_checkpoint,