vmoras commited on
Commit
59d0659
1 Parent(s): 3136bf7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -33,6 +33,8 @@ def get_polygon(points, image, predictor):
33
  """
34
  points = list(chain.from_iterable(points))
35
 
 
 
36
  masks, _, _ = predictor.predict(
37
  box=input_box[None, :],
38
  multimask_output=False,
 
33
  """
34
  points = list(chain.from_iterable(points))
35
 
36
+ input_box = np.array(points)
37
+
38
  masks, _, _ = predictor.predict(
39
  box=input_box[None, :],
40
  multimask_output=False,