Spaces:
Running
on
Zero
Running
on
Zero
Update sam2_mask.py
Browse files- 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)
|