Update gradio_app.py

gradio_app.py

@@ -225,6 +225,26 @@ def inference(id_image, hair_image):
         bald_head,
     ) = _import_inference_bits()
 
+    # Disable StyleGAN2 custom CUDA ops to avoid JIT compiling (needs ninja/NVCC).
+    # ZeroGPU 下建议走纯 PyTorch 引用实现，避免扩展编译失败。
+    try:
+        from HairMapper.styleGAN2_ada_model.stylegan2_ada.torch_utils.ops import bias_act as _bias_act
+        _bias_act.USING_CUDA_TO_SPEED_UP = False
+        try:
+            from HairMapper.styleGAN2_ada_model.stylegan2_ada.torch_utils.ops import upfirdn2d as _upfirdn2d
+            if hasattr(_upfirdn2d, 'USING_CUDA_TO_SPEED_UP'):
+                _upfirdn2d.USING_CUDA_TO_SPEED_UP = False
+        except Exception:
+            pass
+        try:
+            from HairMapper.styleGAN2_ada_model.stylegan2_ada.torch_utils.ops import filtered_lrelu as _fl
+            if hasattr(_fl, 'USING_CUDA_TO_SPEED_UP'):
+                _fl.USING_CUDA_TO_SPEED_UP = False
+        except Exception:
+            pass
+    except Exception:
+        pass
+
     os.makedirs("gradio_inputs", exist_ok=True)
     os.makedirs("gradio_outputs", exist_ok=True)