Update app.py
Browse files
app.py
CHANGED
|
@@ -17,6 +17,8 @@ cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.7
|
|
| 17 |
cfg.MODEL.WEIGHTS = model_zoo.get_checkpoint_url(
|
| 18 |
"COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml"
|
| 19 |
)
|
|
|
|
|
|
|
| 20 |
predictor = DefaultPredictor(cfg)
|
| 21 |
|
| 22 |
# --- Função de inferência ---
|
|
|
|
| 17 |
cfg.MODEL.WEIGHTS = model_zoo.get_checkpoint_url(
|
| 18 |
"COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml"
|
| 19 |
)
|
| 20 |
+
cfg.MODEL.DEVICE = "cpu"
|
| 21 |
+
|
| 22 |
predictor = DefaultPredictor(cfg)
|
| 23 |
|
| 24 |
# --- Função de inferência ---
|