update

app.py

@@ -175,6 +175,7 @@ def encode_frames_to_video(frames_dir: str, out_path: str, fps: float):
 
 # ----------------- MAIN PIPELINE (CUDA-only) -----------------
 def run_pipeline_cuda(bw_video_path: str, ref_image_path: str, user_config: dict, debug_shapes: bool) -> str:
+    print(bw_video_path, ref_image_path)
     if not torch.cuda.is_available():
         raise RuntimeError("未检测到 GPU。此 Space 仅支持 ZeroGPU (CUDA)。")
 
@@ -203,7 +204,7 @@ def run_pipeline_cuda(bw_video_path: str, ref_image_path: str, user_config: dict
 
     # 3) 配置（字段与 main.py 一致；值从 UI 合并）
     default_config = {
-        "FirstFrameIsNotExemplar": False,
+        "FirstFrameIsNotExemplar": True,
         "d16_batch_path": "input",   # parity only
         "ref_path": "ref",           # parity only
         "output": "result",          # parity only
@@ -291,6 +292,8 @@ def run_pipeline_cuda(bw_video_path: str, ref_image_path: str, user_config: dict
                 if not config['FirstFrameIsNotExemplar']:
                     msk = msk[:, 1:3, :, :] if msk is not None else None
 
+                print(rgb.shape, msk.shape)
+
                 info = data['info']
                 frame = info['frame'][0]
                 shape = info['shape']