oxkitsune commited on
Commit
6f20d0a
·
1 Parent(s): b2ee272

load model once

Browse files
Files changed (1) hide show
  1. app.py +0 -11
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