download pretrained models

app.py

@@ -1,6 +1,7 @@
 import rerun as rr
 import rerun.blueprint as rrb
 import depth_pro
+import subprocess
 
 import torch
 import cv2
@@ -8,6 +9,16 @@ from pathlib import Path
 import gradio as gr
 from gradio_rerun import Rerun
 
+# Run the script to get pretrained models
+subprocess.run(["bash", "get_pretrained_models.sh"])
+
+device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+
+# Load model and preprocessing transform
+model, transform = depth_pro.create_model_and_transforms()
+model = model.to(device)
+model.eval()
+
 
 @rr.thread_local_stream("rerun_example_ml_depth_pro")
 def run_ml_depth_pro(model, transform, frames):

get_pretrained_models.sh

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+#
+# For licensing see accompanying LICENSE file.
+# Copyright (C) 2024 Apple Inc. All Rights Reserved.
+#
+mkdir -p checkpoints
+# Place final weights here:
+wget https://ml-site.cdn-apple.com/models/depth-pro/depth_pro.pt -P checkpoints