jatamura commited on
Commit
d6de968
·
verified ·
1 Parent(s): 4b4ca60

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -128,8 +128,7 @@ def server(input, output, session: Session):
128
  # Run prediction with original BGR image
129
  prediction = predictor(im)
130
 
131
- if input.cls_agnostic_nms():
132
- prediction = apply_nms(prediction, input.mask(), input.threshold())
133
 
134
  classes = prediction["instances"].pred_classes.tolist()
135
 
 
128
  # Run prediction with original BGR image
129
  prediction = predictor(im)
130
 
131
+ prediction = apply_nms(prediction, input.mask(), input.threshold())
 
132
 
133
  classes = prediction["instances"].pred_classes.tolist()
134