Update app.py

app.py

@@ -56,17 +56,18 @@ def process_image(image: Image.Image):
     far_blur  = cv2.GaussianBlur(img_np, (81,81), 20)
     
     # invert so 0 -> near, 1 -> far
-    inv_d    = 1.0 - depth_norm
-    alpha    = inv_d[...,None]
+    inv_d = 1.0 - depth_norm
+    alpha = inv_d[...,None]
     
     # now near (inv_d≈1) stays sharp, far (inv_d≈0) becomes far_blur
     combined = img_np * alpha + far_blur * (1.0 - alpha)
     lens_blur = Image.fromarray(np.clip(combined,0,255).astype(np.uint8))
 
-    # 6) composite to keep foreground sharp, invert the binary segmentation mask:
-    bg_mask   = Image.fromarray((mask==0).astype(np.uint8)*255)
+    # 6) composite to keep foreground sharp
+    mask_np = np.array(mask)
+    inv_mask = (mask_np == 0).astype(np.uint8) * 255
+    bg_mask  = Image.fromarray(inv_mask)
     final_lens_blur = Image.composite(lens_blur, image, mask=bg_mask)
-    return image, output_blur, final_lens_blur
 
 iface = gr.Interface(
     fn=process_image,