Upload folder using huggingface_hub

app.py

@@ -17,7 +17,7 @@ def prepare_image(img_path):
     heatmap_n_pose = draw_pose(heatmap_pil,body_keypoints,0.5) #PIL.Image
     
     # area in bounding_box below head (rectangular)
-    face_y1, face_h = face_bboxes[0][1], face_bboxes[0][2] #xywh of first face
+    face_y1, face_h = face_bboxes[0][1], face_bboxes[0][3] #xywh of first face
     face_y2 = face_y1 + face_h
     body_box = body_bboxes.cpu().numpy().xyxy[0]
     body_box[1] = face_y2 #shift below head