fix

app.py

@@ -230,7 +230,6 @@ def load_pipeline_cpu() -> Tuple[object, object, int]:
 
     logger.info(f"repo_root = {repo_root}")
     logger.info(f"qwen_root = {qwen_root}")
-    logger.info(f"torch.cuda.is_available (startup) = {torch.cuda.is_available()}")
 
     exp_cfg = OmegaConf.load(repo_root / "config.yaml")
     exp_cfg = OmegaConf.to_container(exp_cfg, resolve=True)
@@ -281,6 +280,14 @@ def run_edit(
     # 1) 取 CPU 缓存
     model_cpu, scheduler, target_sr = load_pipeline_cpu()
 
+    try:
+        if not torch.cuda.is_available():
+            return None, "Error: ZeroGPU did not allocate CUDA. Please retry or restart Space."
+
+    except Exception as e:
+        logger.exception("run_edit failed")
+        return None, f"Error: {type(e).__name__}: {e}"
+
     # 2) ZeroGPU 进入 GPU 区域后，cuda 才会 available
     if not torch.cuda.is_available():
         return None, "Error: ZeroGPU did not allocate CUDA. Please retry or check Space hardware."