Spaces:
Build error
Build error
minor
Browse files- sam2_mask.py +2 -2
sam2_mask.py
CHANGED
|
@@ -170,7 +170,7 @@ def create_sam2_tab():
|
|
| 170 |
with gr.Row():
|
| 171 |
point_type = gr.Radio(label="point type", choices=["include", "exclude"], value="include")
|
| 172 |
clear_points_btn = gr.Button("Clear Points")
|
| 173 |
-
checkpoint = gr.Dropdown(label="Checkpoint", choices=["tiny", "small", "base-plus", "large"], value="tiny")
|
| 174 |
submit_btn = gr.Button("Submit")
|
| 175 |
with gr.Column():
|
| 176 |
output_result = gr.Image()
|
|
@@ -195,7 +195,7 @@ def create_sam2_tab():
|
|
| 195 |
)
|
| 196 |
submit_btn.click(
|
| 197 |
fn=sam_process,
|
| 198 |
-
inputs=[input_image,
|
| 199 |
outputs=[output_result, output_result_mask]
|
| 200 |
)
|
| 201 |
return input_image, points_map, output_result_mask
|
|
|
|
| 170 |
with gr.Row():
|
| 171 |
point_type = gr.Radio(label="point type", choices=["include", "exclude"], value="include")
|
| 172 |
clear_points_btn = gr.Button("Clear Points")
|
| 173 |
+
# checkpoint = gr.Dropdown(label="Checkpoint", choices=["tiny", "small", "base-plus", "large"], value="tiny")
|
| 174 |
submit_btn = gr.Button("Submit")
|
| 175 |
with gr.Column():
|
| 176 |
output_result = gr.Image()
|
|
|
|
| 195 |
)
|
| 196 |
submit_btn.click(
|
| 197 |
fn=sam_process,
|
| 198 |
+
inputs=[input_image, tracking_points, trackings_input_label],
|
| 199 |
outputs=[output_result, output_result_mask]
|
| 200 |
)
|
| 201 |
return input_image, points_map, output_result_mask
|