update code

app.py

@@ -48,7 +48,7 @@ def apply_blur(image, blur_type, blur_strength, depth_threshold):
             align_corners=False,
         )
 
-        mask = prediction[0, 0, :, :].detach().cpu().numpy() < depth_threshold
+        mask = prediction[0, 0, :, :].detach().cpu().numpy() > depth_threshold
         mask = mask.astype(np.uint8)
 
     # Invert mask using cv2
@@ -76,7 +76,7 @@ iface = gr.Interface(
         gr.Image(label="Input Image"),
         gr.Radio(["Gaussian", "Lens"], label="Blur Type", value="Gaussian"),
         gr.Slider(1, 30, value=15, step=1, label="Blur Strength"),
-        gr.Slider(-10, 10, value=-4, step=0.1, label="Depth Threshold")
+        gr.Slider(-20, 20, value=-4, step=0.1, label="Depth Threshold")
     ],
     outputs=gr.Image(label="Output Image"),
     title="Image Segmentation and Blurring",