Hugging Face's logo

Spaces:
SIH
/
building-segmentation
Runtime error

App Files Community
hlydecker commited on
Commit
9810614
•
1 Parent(s): deed469

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -28,8 +28,7 @@ cfg = get_cfg()
28
  cfg.merge_from_file("model_weights/buildings_poc_cfg.yml")
29
  cfg.MODEL.DEVICE='cpu'
30
  cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.35
31
- cfg.MODEL.WEIGHTS = "model_weights/model_final.pth
32
- "
33
  cfg.MODEL.ROI_HEADS.NUM_CLASSES = 8
34
  predictor = DefaultPredictor(cfg)
35
 
 
28
  cfg.merge_from_file("model_weights/buildings_poc_cfg.yml")
29
  cfg.MODEL.DEVICE='cpu'
30
  cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.35
31
+ cfg.MODEL.WEIGHTS = "model_weights/model_final.pth"
 
32
  cfg.MODEL.ROI_HEADS.NUM_CLASSES = 8
33
  predictor = DefaultPredictor(cfg)
34