update

dg845 · Aug 4, 2023 · e24691a · e24691a
1 parent 98a88ca
commit e24691a
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -35,7 +35,7 @@ img = Image.open(BytesIO(response.content)).convert("RGB").resize((512, 512))
 #  "lineart_coarse", "lineart_realistic", "mediapipe_face", "mlsd", "normal_bae", "normal_midas",
 #  "openpose", "openpose_face", "openpose_faceonly", "openpose_full", "openpose_hand",
 #  "scribble_hed, "scribble_pidinet", "shuffle", "softedge_hed", "softedge_hedsafe",
-#  "softedge_pidinet", "softedge_pidsafe", "dwpose"]
+#  "softedge_pidinet", "softedge_pidsafe"]
 processor_id = 'scribble_hed'
 processor = Processor(processor_id)
 

diff --git a/src/controlnet_aux/__init__.py b/src/controlnet_aux/__init__.py
@@ -14,4 +14,5 @@
 from .canny import CannyDetector
 from .mediapipe_face import MediapipeFaceDetector
 from .segment_anything import SamDetector
-from .shuffle import ContentShuffleDetector
+from .shuffle import ContentShuffleDetector
+from .dwpose import DWposeDetector
diff --git a/src/controlnet_aux/processor.py b/src/controlnet_aux/processor.py
@@ -11,7 +11,7 @@
                             LeresDetector, LineartAnimeDetector,
                             LineartDetector, MediapipeFaceDetector,
                             MidasDetector, MLSDdetector, NormalBaeDetector,
-                            OpenposeDetector, PidiNetDetector, ZoeDetector)
+                            OpenposeDetector, PidiNetDetector, ZoeDetector,)
 
 LOGGER = logging.getLogger(__name__)