Hugging Face's logo

Spaces:
SIH
/
aerial-segmentation-model-selection
Runtime error

App Files Community
xinwei89 commited on
Commit
d4def87
·
1 Parent(s): 74ee8cd

fix typo

Browse files
Files changed (1) hide show
  1. backend.py +1 -1
backend.py CHANGED
@@ -37,7 +37,7 @@ tree_predictor = DefaultPredictor(tree_cfg)
37
 
38
  # Model for buildings
39
  building_cfg = get_cfg()
40
- building_cfg.merge_from_file("building_model_weight/building_poc_cfg.yml")
41
  building_cfg.MODEL.DEVICE='cpu'
42
  building_cfg.MODEL.WEIGHTS = "building_model_weight/model_final.pth"
43
  building_cfg.MODEL.ROI_HEADS.NUM_CLASSES = 8
 
37
 
38
  # Model for buildings
39
  building_cfg = get_cfg()
40
+ building_cfg.merge_from_file("building_model_weight/buildings_poc_cfg.yml")
41
  building_cfg.MODEL.DEVICE='cpu'
42
  building_cfg.MODEL.WEIGHTS = "building_model_weight/model_final.pth"
43
  building_cfg.MODEL.ROI_HEADS.NUM_CLASSES = 8