Update app.py

app.py

@@ -14,12 +14,13 @@ def depthMap(imgL,imgR):
   stereoMatcher.setSpeckleRange(16)
   stereoMatcher.setSpeckleWindowSize(45)
   disparity = stereoMatcher.compute(imgL,imgR)
-  depth = (0.1 * 500) / disparity
+  depth_map = np.zeros_like(disparity, dtype=np.float32)
+  depth_map[disparity_map > 0] = 0.1 * 500 / disparity[disparity > 0]
   gray = plt.get_cmap('gray')
   disparity = disparity - np.min(disparity)
   disparity = disparity / np.max(disparity)
   disparity = gray(disparity)[:, :, :3]
-  return depth
+  return depth_map
 
 leftCam = gr.inputs.Image(type="numpy")
 RightCam = gr.inputs.Image(type="numpy")