Update app.py

app.py

@@ -60,7 +60,7 @@ def get_depth_image(image):
     depth_estimator = pipeline('depth-estimation')
     pil_image = depth_estimator(pil_image)['depth']
     pil_image = np.array(pil_image)
-    pil_image = pil_image[:, None]
+    pil_image = pil_image[:, :, None]
     pil_image = np.concatenate([pil_image, pil_image, pil_image], axis=2)
     return PIL.Image.fromarray(pil_image)