Update app.py

app.py

@@ -160,11 +160,14 @@ def convert_from_cv2_to_image(img: np.ndarray) -> Image:
 def convert_from_image_to_cv2(img: Image) -> np.ndarray:
     return cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR)
 
-def draw_kps(w,h, kps, color_list=[(255,0,0), (0,255,0), (0,0,255), (255,255,0), (255,0,255)]):
+
+def draw_kps(image_pil, kps, color_list=[(255,0,0), (0,255,0), (0,0,255), (255,255,0), (255,0,255)]):
     
-    stickwidth = 8
+    stickwidth = 4
     limbSeq = np.array([[0, 2], [1, 2], [3, 2], [4, 2]])
+    kps = np.array(kps)
 
+    w, h = image_pil.size
     out_img = np.zeros([h, w, 3])
 
     for i in range(len(limbSeq)):
@@ -184,7 +187,6 @@ def draw_kps(w,h, kps, color_list=[(255,0,0), (0,255,0), (0,0,255), (255,255,0),
         x, y = kp
         out_img = cv2.circle(out_img.copy(), (int(x), int(y)), 10, color, -1)
 
-    # out_img = out_img.astype(np.uint8)
     out_img_pil = PIL.Image.fromarray(out_img.astype(np.uint8))
     return out_img_pil