Spaces:

SIH
/

building-segmentation

Runtime error

hlydecker commited on Nov 6, 2023

Commit

e679284

•

1 Parent(s): 05d4fcf

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -25,7 +25,7 @@ from detectron2.data import MetadataCatalog, DatasetCatalog
 from detectron2.checkpoint import DetectionCheckpointer
 cfg = get_cfg()
-cfg.merge_from_file("model_weights/buildingsv1_cfg.yaml")
 cfg.MODEL.DEVICE='cpu'
 cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.35
 cfg.MODEL.WEIGHTS = "model_weights/buildingsv1_best.pth"

 from detectron2.checkpoint import DetectionCheckpointer
 cfg = get_cfg()
+cfg.merge_from_file("model_weights/buildings_poc_cfg.yml")
 cfg.MODEL.DEVICE='cpu'
 cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.35
 cfg.MODEL.WEIGHTS = "model_weights/buildingsv1_best.pth"