Spaces:
Running
on
Zero
Running
on
Zero
load model once
Browse files
app.py
CHANGED
@@ -69,17 +69,6 @@ def run_ml_depth_pro(model, transform, frames):
|
|
69 |
|
70 |
|
71 |
video_path = Path("hd-cat.mp4")
|
72 |
-
device = torch.device(
|
73 |
-
"cuda"
|
74 |
-
if torch.cuda.is_available()
|
75 |
-
else "mps"
|
76 |
-
if torch.backends.mps.is_available()
|
77 |
-
else "cpu"
|
78 |
-
)
|
79 |
-
|
80 |
-
# Load model and preprocessing transform
|
81 |
-
model, transform = depth_pro.create_model_and_transforms(device=device)
|
82 |
-
model.eval()
|
83 |
|
84 |
|
85 |
# Load video
|
|
|
69 |
|
70 |
|
71 |
video_path = Path("hd-cat.mp4")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
72 |
|
73 |
|
74 |
# Load video
|