Update depth.py

depth.py

@@ -1,6 +1,7 @@
 from zoedepth.utils.misc import colorize, save_raw_16bit
 import torch
 import gradio as gr
+import spaces
 from PIL import Image
 import numpy as np
 from functools import partial
@@ -24,6 +25,7 @@ def process_image(model, image: Image.Image):
     processed_array = save_raw_16bit(colorize(out)[:, :, 0])
     return Image.fromarray(processed_array)
 
+@spaces.GPU
 def depth_interface(model, device):
     with gr.Row():
         inputs=gr.Image(label="Input Image", type='pil')  # Input is an image