update

app.py

@@ -29,8 +29,8 @@ def get_openpose_detector():
 def get_dwpose_detector():
     """Get or create DWPose detector."""
     if "dwpose" not in _model_cache:
-        from easy_dwpose import DWposeDetector
-        _model_cache["dwpose"] = DWposeDetector(device=DEVICE)
+        from controlnet_aux import DWposeDetector
+        _model_cache["dwpose"] = DWposeDetector.from_pretrained("yolox_l.onnx", "dw-ll_ucoco_384.onnx")
     return _model_cache["dwpose"]
 
 
@@ -85,22 +85,17 @@ def process_with_dwpose(
     """Process image using DWPose detector."""
     detector = get_dwpose_detector()
     
-    # Resize image to detect_resolution while maintaining aspect ratio
-    orig_w, orig_h = image.size
-    scale = detect_resolution / max(orig_w, orig_h)
-    new_w, new_h = int(orig_w * scale), int(orig_h * scale)
-    resized_image = image.resize((new_w, new_h), Image.Resampling.LANCZOS)
-    
+    # controlnet-aux DWposeDetector API
     result = detector(
-        resized_image,
-        output_type="pil",
-        include_hands=detect_hand,
-        include_face=detect_face
+        image,
+        detect_resolution=detect_resolution,
+        image_resolution=detect_resolution,
+        include_hand=detect_hand,
+        include_face=detect_face,
+        include_body=True,
+        output_type="pil"
     )
     
-    # Resize back to original size
-    result = result.resize((orig_w, orig_h), Image.Resampling.LANCZOS)
-    
     return result, None
 
 

requirements.txt

@@ -1,10 +1,7 @@
 gradio>=4.0.0
 controlnet-aux>=0.0.9
-onnxruntime>=1.16.0
-easy-dwpose --no-deps
 torch
 torchvision
 Pillow
 numpy
 opencv-python-headless
-huggingface_hub