depth issue

gradio_app.py

@@ -35,7 +35,7 @@ def download_models():
         print("✅ Pretrained models already exist.")
 
 
-def visualize_depth_npy_as_video(npy_file):
+def visualize_depth_npy_as_video(npy_file, fps):
     # Load .npy file
     depth_np = np.load(npy_file.name)  # Shape: [T, 1, H, W]
     tensor = torch.from_numpy(depth_np)
@@ -44,7 +44,7 @@ def visualize_depth_npy_as_video(npy_file):
     # Prepare video writer
     video_path = "depth_video_preview.mp4"
     fourcc = cv2.VideoWriter_fourcc(*'mp4v')
-    out = cv2.VideoWriter(video_path, fourcc, 10, (W, H))  # 10 FPS
+    out = cv2.VideoWriter(video_path, fourcc, fps, (W, H))  # 10 FPS
 
     for i in range(T):
         frame = tensor[i, 0].numpy()
@@ -72,7 +72,7 @@ def get_anchor_video(video_path, fps, num_frames, target_pose, mode,
     video_output_path = f"{output_dir}/masked_videos/output.mp4"
     captions_text_file = f"{output_dir}/captions/output.txt"
     depth_file = f"{output_dir}/depth/output.npy"
-    depth_video_path = visualize_depth_video(depth_file)
+    depth_video_path = visualize_depth_npy_as_video(depth_file, fps)
 
 
     if video_path: