orhir commited on
Commit
ee7caad
·
verified ·
1 Parent(s): 6e6b7e1

Update gradio_utils/utils.py

Browse files
Files changed (1) hide show
  1. gradio_utils/utils.py +2 -1
gradio_utils/utils.py CHANGED
@@ -301,7 +301,8 @@ def select_skeleton(global_state,
301
  if global_state["curr_type_point"] == "end":
302
  prev_point_idx = global_state["prev_point_idx"]
303
  prev_point = pts_list[prev_point_idx]
304
- points = [prev_point, xy]
 
305
  image_draw = draw_limbs_on_image(image_raw,
306
  points
307
  )
 
301
  if global_state["curr_type_point"] == "end":
302
  prev_point_idx = global_state["prev_point_idx"]
303
  prev_point = pts_list[prev_point_idx]
304
+ current_point = pts_list[closest_point_idx]
305
+ points = [prev_point, current_point]
306
  image_draw = draw_limbs_on_image(image_raw,
307
  points
308
  )