fix

app.py

@@ -213,9 +213,12 @@ def render_from_pointcloud(rgb_frames, depth_frames, intrinsics, original_extrin
 
 @spaces.GPU
 def run_spatial_tracker(video_tensor):
+    if tracker_model.spatrack.base_model is None:
+        tracker_model.spatrack.base_model = vggt4track_model
+
     video_input = preprocess_image(video_tensor)[None].cuda()
     with torch.no_grad():
-        with torch.cuda.amp.autocast(dtype=torch.bfloat16):
+        with torch.amp.autocast('cuda', dtype=torch.bfloat16):
             predictions = vggt4track_model(video_input / 255)
             extrinsic, intrinsic = predictions["poses_pred"], predictions["intrs"]
             depth_map, depth_conf = predictions["points_map"][...,