Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
| 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 |
|