CharlieAmalet commited on
Commit
bbe8871
·
verified ·
1 Parent(s): 968b343

Update depth.py

Browse files
Files changed (1) hide show
  1. depth.py +1 -1
depth.py CHANGED
@@ -17,6 +17,7 @@ def save_raw_16bit(depth, fpath="raw.png"):
17
  return depth
18
 
19
  # Your image processing function
 
20
  def process_image(model, image: Image.Image):
21
  image = image.convert("RGB")
22
 
@@ -25,7 +26,6 @@ def process_image(model, image: Image.Image):
25
  processed_array = save_raw_16bit(colorize(out)[:, :, 0])
26
  return Image.fromarray(processed_array)
27
 
28
- @spaces.GPU
29
  def depth_interface(model, device):
30
  with gr.Row():
31
  inputs=gr.Image(label="Input Image", type='pil') # Input is an image
 
17
  return depth
18
 
19
  # Your image processing function
20
+ @spaces.GPU(enable_queue=True)
21
  def process_image(model, image: Image.Image):
22
  image = image.convert("RGB")
23
 
 
26
  processed_array = save_raw_16bit(colorize(out)[:, :, 0])
27
  return Image.fromarray(processed_array)
28
 
 
29
  def depth_interface(model, device):
30
  with gr.Row():
31
  inputs=gr.Image(label="Input Image", type='pil') # Input is an image