Hugging Face's logo

Spaces:
SIH
/
tree-segmentation
Runtime error

App Files Community
hlydecker commited on
Commit
bca0412
·
1 Parent(s): 5330255

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ cfg.merge_from_file("model_weights/tree_cfg.yml")
29
  cfg.MODEL.DEVICE='cpu'
30
  cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.25
31
  cfg.MODEL.WEIGHTS = "model_weights/tree_model.pth"
32
- cfg.MODEL.ROI_HEADS.NUM_CLASSES = 8
33
  predictor = DefaultPredictor(cfg)
34
 
35
  def segment_buildings(im):
 
29
  cfg.MODEL.DEVICE='cpu'
30
  cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.25
31
  cfg.MODEL.WEIGHTS = "model_weights/tree_model.pth"
32
+ cfg.MODEL.ROI_HEADS.NUM_CLASSES = 2
33
  predictor = DefaultPredictor(cfg)
34
 
35
  def segment_buildings(im):