Update app.py

app.py

@@ -232,25 +232,26 @@ def build_args_list_for_test(d16_batch_path: str,
             args.extend([flag, str(v)])
     return args
 
-# ===== 新增：ZeroGPU 后按需安装 Pytorch-Correlation-extension =====
-
-  def ensure_correlation_extension_installed():
-      """Install spatial_correlation_sampler at runtime (needs CUDA)."""
-      try:
-          import spatial_correlation_sampler
-          return
-      except ImportError:
-          pass
-      print("[INFO] Installing spatial-correlation-sampler via pip (needs CUDA)...")
-      subprocess.run(
-          [sys.executable, "-m", "pip", "install", "spatial-correlation-sampler"],
-          check=True
-      )
-      importlib.invalidate_caches()
-      import spatial_correlation_sampler  # verify
-      print("[INFO] spatial-correlation-sampler installed successfully.")
-      
+# ===== ZeroGPU: install spatial-correlation-sampler at runtime =====
+
+def ensure_correlation_extension_installed():
+    """Install spatial_correlation_sampler at runtime (needs CUDA)."""
+    try:
+        import spatial_correlation_sampler
+        return
+    except ImportError:
+        pass
+    print("[INFO] Installing spatial-correlation-sampler via pip (needs CUDA)...")
+    subprocess.run(
+        [sys.executable, "-m", "pip", "install", "spatial-correlation-sampler"],
+        check=True
+    )
+    importlib.invalidate_caches()
+    import spatial_correlation_sampler  # verify
+    print("[INFO] spatial-correlation-sampler installed successfully.")
+
 # ----------------- GRADIO HANDLER -----------------
+
 @spaces.GPU(duration=120)   # 确保 CUDA 初始化在此函数体内
 def gradio_infer(
     debug_shapes,