Spaces:
Sleeping
Sleeping
test
Browse files
app.py
CHANGED
|
@@ -212,9 +212,12 @@ def run_spatial_tracker(video_tensor: torch.Tensor):
|
|
| 212 |
Dictionary containing tracking results
|
| 213 |
"""
|
| 214 |
# Run VGGT to get depth and camera poses
|
|
|
|
|
|
|
|
|
|
|
|
|
| 215 |
video_input = preprocess_image(video_tensor)[None].cuda()
|
| 216 |
|
| 217 |
-
vggt4track_model.eval()
|
| 218 |
vggt4track_model = vggt4track_model.to("cuda")
|
| 219 |
|
| 220 |
with torch.no_grad():
|
|
|
|
| 212 |
Dictionary containing tracking results
|
| 213 |
"""
|
| 214 |
# Run VGGT to get depth and camera poses
|
| 215 |
+
global vggt4track_model
|
| 216 |
+
global tracker_model
|
| 217 |
+
global wan_pipeline
|
| 218 |
+
|
| 219 |
video_input = preprocess_image(video_tensor)[None].cuda()
|
| 220 |
|
|
|
|
| 221 |
vggt4track_model = vggt4track_model.to("cuda")
|
| 222 |
|
| 223 |
with torch.no_grad():
|