fffiloni commited on
Commit
01bbb6c
·
verified ·
1 Parent(s): a37e840

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -120,7 +120,7 @@ def sam_process(input_image, tracking_points, trackings_input_label):
120
  masks, scores, logits = predictor.predict(
121
  point_coords=input_point,
122
  point_labels=input_label,
123
- multimask_output=True,
124
  )
125
  sorted_ind = np.argsort(scores)[::-1]
126
  masks = masks[sorted_ind]
@@ -149,7 +149,7 @@ with gr.Blocks() as demo:
149
 
150
  input_image.upload(preprocess_image, input_image, [first_frame_path, tracking_points, trackings_input_label])
151
 
152
- input_image.select(get_point, [tracking_points, trackings_input_label, first_frame_path], [tracking_points, trackings_input_label, points_map])
153
 
154
 
155
  submit_btn.click(
 
120
  masks, scores, logits = predictor.predict(
121
  point_coords=input_point,
122
  point_labels=input_label,
123
+ multimask_output=False,
124
  )
125
  sorted_ind = np.argsort(scores)[::-1]
126
  masks = masks[sorted_ind]
 
149
 
150
  input_image.upload(preprocess_image, input_image, [first_frame_path, tracking_points, trackings_input_label])
151
 
152
+ points_map.select(get_point, [tracking_points, trackings_input_label, first_frame_path], [tracking_points, trackings_input_label, points_map])
153
 
154
 
155
  submit_btn.click(