Update app.py

app.py

@@ -25,13 +25,13 @@ from detectron2.data import MetadataCatalog, DatasetCatalog
 from detectron2.checkpoint import DetectionCheckpointer
 
 cfg = get_cfg()
-cfg.merge_from_file("model_weights/lczs_cfg.yaml")
+cfg.merge_from_file("model_weights/gsu-lcz-sample-model_cfg.yaml")
 cfg.MODEL.DEVICE='cpu'
 cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.25 
 #cfg.MODEL.WEIGHTS = "model_weights/model_final.pth"  
 #cfg.MODEL.ROI_HEADS.NUM_CLASSES = 8
-cfg.MODEL.WEIGHTS = "model_weights/lczs_v2.pth"  
-cfg.MODEL.ROI_HEADS.NUM_CLASSES = 14
+cfg.MODEL.WEIGHTS = "model_weights/gsu-lcz-sample-model_best.pth"  
+cfg.MODEL.ROI_HEADS.NUM_CLASSES = 13
 predictor = DefaultPredictor(cfg)
 
 def segment(im):