Update app.py

app.py

@@ -27,13 +27,12 @@ from detectron2.checkpoint import DetectionCheckpointer
 cfg = get_cfg()
 cfg.merge_from_file("model_weights/treev1_cfg.yaml")
 cfg.MODEL.DEVICE='cpu'
-cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.75
 cfg.MODEL.WEIGHTS = "model_weights/treev1_best.pth"  
 cfg.MODEL.ROI_HEADS.NUM_CLASSES = 2
 
 def segment_image(im):
 
-    cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.75
+    cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.25
     predictor = DefaultPredictor(cfg)
     im = np.array(im)
     outputs = predictor(im)