hlydecker commited on
Commit
82d85d1
1 Parent(s): 9964a9f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ from detectron2.data import MetadataCatalog, DatasetCatalog
25
  from detectron2.checkpoint import DetectionCheckpointer
26
 
27
  cfg = get_cfg()
28
- cfg.merge_from_file("model_weights/treev1_cfg.yml")
29
  cfg.MODEL.DEVICE='cpu'
30
  cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.75
31
  cfg.MODEL.WEIGHTS = "model_weights/treev1_best.pth"
 
25
  from detectron2.checkpoint import DetectionCheckpointer
26
 
27
  cfg = get_cfg()
28
+ cfg.merge_from_file("model_weights/treev1_cfg.yaml")
29
  cfg.MODEL.DEVICE='cpu'
30
  cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.75
31
  cfg.MODEL.WEIGHTS = "model_weights/treev1_best.pth"