Spaces:
Runtime error
Runtime error
RamAnanth1
commited on
Commit
•
bc521ff
1
Parent(s):
8f3f6b7
Change cmap to magma
Browse files
app.py
CHANGED
@@ -15,7 +15,7 @@ zoe = model_zoe_n.to(DEVICE)
|
|
15 |
|
16 |
def process_image(image):
|
17 |
depth = zoe.infer_pil(image) # as numpy
|
18 |
-
colored_depth = colorize(depth)
|
19 |
return colored_depth
|
20 |
|
21 |
interface = gr.Interface(fn=process_image,
|
|
|
15 |
|
16 |
def process_image(image):
|
17 |
depth = zoe.infer_pil(image) # as numpy
|
18 |
+
colored_depth = colorize(depth, cmap = 'magma_r')
|
19 |
return colored_depth
|
20 |
|
21 |
interface = gr.Interface(fn=process_image,
|