abreza commited on
Commit
aa4c733
·
1 Parent(s): beef5e1
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -117,9 +117,6 @@ print("✅ Tracking models loaded successfully!")
117
 
118
  gr.set_static_paths(paths=[Path.cwd().absolute()/"_viz"])
119
 
120
- # --- YOUR ORIGINAL FUNCTIONS (generate_camera_trajectory, render_from_pointcloud, run_spatial_tracker) ---
121
- # [Keeping these exactly as provided in your snippet]
122
-
123
 
124
  def generate_camera_trajectory(num_frames: int, movement_type: str, base_intrinsics: np.ndarray, scene_scale: float = 1.0) -> tuple:
125
  speed = scene_scale * 0.02
@@ -213,6 +210,8 @@ def render_from_pointcloud(rgb_frames, depth_frames, intrinsics, original_extrin
213
 
214
  @spaces.GPU
215
  def run_spatial_tracker(video_tensor):
 
 
216
  if tracker_model.spatrack.base_model is None:
217
  tracker_model.spatrack.base_model = vggt4track_model
218
 
 
117
 
118
  gr.set_static_paths(paths=[Path.cwd().absolute()/"_viz"])
119
 
 
 
 
120
 
121
  def generate_camera_trajectory(num_frames: int, movement_type: str, base_intrinsics: np.ndarray, scene_scale: float = 1.0) -> tuple:
122
  speed = scene_scale * 0.02
 
210
 
211
  @spaces.GPU
212
  def run_spatial_tracker(video_tensor):
213
+ if not hasattr(vggt4track_model, "infer"):
214
+ vggt4track_model.infer = lambda x: vggt4track_model(x)
215
  if tracker_model.spatrack.base_model is None:
216
  tracker_model.spatrack.base_model = vggt4track_model
217