get pose coords

main.py

@@ -48,7 +48,7 @@ from src.pipelines.PCDMs_pipeline import PCDMsPipeline
 
 
 import spaces
-from .libs.easy_dwpose import DWposeDetector
+from libs.easy_dwpose import DWposeDetector
 from PIL import Image
 import cv2
 import os
@@ -140,7 +140,7 @@ def getThumbnails(imgs):
 # Pose detection ==============================================================================================
 
 def load_models():
-    dwpose = DWposeDetector(device="cpu")
+    dwpose = DWposeDetector(device="cuda")
     rembg_session = rembg.new_session("u2netp")
 
     pcdms_model = hf_hub_download(repo_id="acmyu/PCDMs", filename="pcdms_ckpt.pt")
@@ -212,7 +212,7 @@ def get_pose(img, dwpose, outfile, crop=False):
     #skeleton = dwpose(pil_image, output_type="np", include_hands=True, include_face=False)
 
     #img.thumbnail((512,512))
-    out_img, _ = dwpose(img, include_hands=True, include_face=False)
+    out_img, pose = dwpose(img, include_hands=True, include_face=False)
 
     #print(pose['bodies'])