Update app.py

app.py

@@ -30,9 +30,9 @@ cfg.MODEL.DEVICE='cpu'
 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.25
+def segment_image(im, confidence_threshold):
+    # cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.25
+    cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = confidence_threshold
     predictor = DefaultPredictor(cfg)
     im = np.array(im)
     outputs = predictor(im)
@@ -46,10 +46,10 @@ def segment_image(im):
     return Image.fromarray(out.get_image()[:, :, ::-1])
 
 # gradio components 
-"""
+
 gr_slider_confidence = gr.inputs.Slider(0,1,.1,.7,
                                         label='Set confidence threshold % for masks')
-"""
+
 # gradio outputs
 inputs = gr.inputs.Image(type="pil", label="Input Image")
 outputs = gr.outputs.Image(type="pil", label="Output Image")
@@ -59,7 +59,7 @@ description = "An instance segmentation demo for identifying trees in aerial ima
 
 # Create user interface and launch
 gr.Interface(segment_image, 
-                inputs = inputs,
+                inputs = [inputs, gr_slider_confidence],
                 outputs = outputs,
                  title = title,
                 description = description).launch(debug=True)