RamAnanth1 commited on
Commit
bc521ff
1 Parent(s): 8f3f6b7

Change cmap to magma

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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,