Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -82,7 +82,7 @@ with gr.Blocks(css=css) as demo:
|
|
| 82 |
|
| 83 |
depth = (depth - depth.min()) / (depth.max() - depth.min()) * 255.0
|
| 84 |
depth = depth.cpu().numpy().astype(np.uint8)
|
| 85 |
-
colored_depth = cv2.applyColorMap(depth, cv2.
|
| 86 |
|
| 87 |
return [(original_image, colored_depth), tmp.name]
|
| 88 |
|
|
|
|
| 82 |
|
| 83 |
depth = (depth - depth.min()) / (depth.max() - depth.min()) * 255.0
|
| 84 |
depth = depth.cpu().numpy().astype(np.uint8)
|
| 85 |
+
colored_depth = cv2.applyColorMap(depth, cv2.COLORMAP_BONE)[:, :, ::-1]
|
| 86 |
|
| 87 |
return [(original_image, colored_depth), tmp.name]
|
| 88 |
|