abreza commited on
Commit
058680b
·
1 Parent(s): 645f131
Files changed (1) hide show
  1. app.py +4 -1
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():