Update app.py

app.py

@@ -275,7 +275,7 @@ def stereo_pipeline(image_pil, divergence_percent, convergence_plane):
     # 5. Convert to numpy
     right_rgb = (right_tensor.squeeze(0).permute(1,2,0).cpu().numpy() * 255).astype(np.uint8)
     right_bgr = cv2.cvtColor(right_rgb, cv2.COLOR_RGB2BGR)
-    mask_np = occlusion_mask.squeeze(0).cpu().numpy()
+    mask_np = occlusion_mask.squeeze().cpu().numpy()
 
     # 6. Two-pass LaMa (perfect edges)
     right_filled_bgr = run_lama_twice(right_bgr, mask_np)