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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -113,7 +113,7 @@ def sam_process(input_image, tracking_points, trackings_input_label):
113
  predictor.set_image(image)
114
 
115
  input_point = np.array(tracking_points.value)
116
- input_label = np.array([1])
117
 
118
  print(predictor._features["image_embed"].shape, predictor._features["image_embed"][-1].shape)
119
 
 
113
  predictor.set_image(image)
114
 
115
  input_point = np.array(tracking_points.value)
116
+ input_label = np.array(trackings_input_label.value)
117
 
118
  print(predictor._features["image_embed"].shape, predictor._features["image_embed"][-1].shape)
119