Spaces:
Running
Running
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
|