KurtLin commited on
Commit
2b08595
·
1 Parent(s): 2bf62e7

Initial Commit

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ def get_coords(evt: gr.SelectData):
19
  def inference(image, input_label):
20
  print(input_label)
21
  predictor.set_image(image)
22
- input_point = np.array([[int(input_label['0']), int(input_label['1'])]])
23
  input_label = np.array([1])
24
  masks, scores, logits = predictor.predict(
25
  point_coords=input_point,
 
19
  def inference(image, input_label):
20
  print(input_label)
21
  predictor.set_image(image)
22
+ input_point = np.array([[int(input_label['label'].split(',')[0]), int(input_label['label'].split(',')[1])]])
23
  input_label = np.array([1])
24
  masks, scores, logits = predictor.predict(
25
  point_coords=input_point,