LPX55 commited on
Commit
b79bee3
·
1 Parent(s): 592fccb
Files changed (1) hide show
  1. 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, checkpoint, tracking_points, trackings_input_label],
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