LPX55 commited on
Commit
93bfbab
·
verified ·
1 Parent(s): 869d6d2

Update sam2_mask.py

Browse files
Files changed (1) hide show
  1. sam2_mask.py +1 -1
sam2_mask.py CHANGED
@@ -13,7 +13,7 @@ from sam2.sam2_image_predictor import SAM2ImagePredictor
13
  def preprocess_image(image):
14
  return image, gr.State([]), gr.State([]), image
15
 
16
- def get_point(point_type, tracking_points, trackings_input_label, original_image, evt):
17
  x, y = evt.index
18
  tracking_points.append((x, y))
19
  trackings_input_label.append(1 if point_type == "include" else 0)
 
13
  def preprocess_image(image):
14
  return image, gr.State([]), gr.State([]), image
15
 
16
+ def get_point(point_type, tracking_points, trackings_input_label, original_image, evt: gr.SelectData):
17
  x, y = evt.index
18
  tracking_points.append((x, y))
19
  trackings_input_label.append(1 if point_type == "include" else 0)