Upload folder using huggingface_hub

app.py

@@ -19,7 +19,7 @@ def prepare_image(img_path):
     # area in bounding_box below head (rectangular)
     face_y1, face_h = face_bboxes[0][1], face_bboxes[0][2] #xywh of first face
     face_y2 = face_y1 + face_h
-    body_box = body_bboxes.numpy().xyxy[0]
+    body_box = body_bboxes.cpu().numpy().xyxy[0]
     body_box[1] = face_y2 #shift below head
     
     # Calculate the coordinates for cropping

pose.py

@@ -49,8 +49,8 @@ def detect_pose(src_img):
     return results
 
 def draw_pose(skeleton_image:Image.Image, keypoints, threshold=0.7, line_width = 3,circle_radius = 6):
-    coords = keypoints.numpy().xy[0]
-    scores = keypoints.data.squeeze()[:,2].numpy()
+    coords = keypoints.cpu().numpy().xy[0]
+    scores = keypoints.data.squeeze()[:,2].cpu().numpy()
     # Create a blank image with the same size as the input image
     # skeleton_image = Image.new("RGB", im.size, "black")  # White background
     draw = ImageDraw.Draw(skeleton_image)