Update depth.py

depth.py

@@ -24,9 +24,9 @@ def process_image(model, image: Image.Image):
     processed_array = save_raw_16bit(colorize(out)[:, :, 0])
     return Image.fromarray(processed_array)
 
-def depth_interface(model):
+def depth_interface(model, device):
     with gr.Row():
         inputs=gr.Image(label="Input Image", type='pil')  # Input is an image
         outputs=gr.Image(label="Depth Map", type='pil')  # Output is also an image
     generate_btn = gr.Button(value="Generate")
-    generate_btn.click(partial(process_image, model), inputs=inputs, outputs=outputs, api_name="generate_depth")
+    generate_btn.click(partial(process_image, model.to(device)), inputs=inputs, outputs=outputs, api_name="generate_depth")