Update app.py

app.py

@@ -25,10 +25,10 @@ from detectron2.data import MetadataCatalog, DatasetCatalog
 from detectron2.checkpoint import DetectionCheckpointer
 
 cfg = get_cfg()
-cfg.merge_from_file("model_weights/tree_cfg.yml")
+cfg.merge_from_file("model_weights/treev1_cfg.yml")
 cfg.MODEL.DEVICE='cpu'
 cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.75
-cfg.MODEL.WEIGHTS = "model_weights/tree_model.pth"  
+cfg.MODEL.WEIGHTS = "model_weights/treev1_best.pth"  
 cfg.MODEL.ROI_HEADS.NUM_CLASSES = 2
 predictor = DefaultPredictor(cfg)