Update app.py

app.py

@@ -227,11 +227,7 @@ def _editing_output_size(input_img: Image.Image, target_pixels: int) -> tuple[in
 
 # Start the guard subprocess before the main model is loaded: Popen forks this
 # process, and forking it after 35GB of weights are resident is far costlier.
-import ncii_guard
 
-print("[startup] starting NCII prompt guard subprocess...")
-ncii_guard.start()
-print(f"[startup] guard ready (threshold {GUARD_THRESHOLD}, edits only)")
 
 print("[startup] loading SenseNova-U1.5-8B-MoT (this may take a few minutes)...")
 sensenova_u1.set_attn_backend("auto")
@@ -254,29 +250,6 @@ import sn_aoti
 print(f"[startup] {sn_aoti.status()}")
 sn_aoti.maybe_load(model)
 
-def _guard_score(prompt: str) -> float | None:
-    """P(ncii) for a prompt, or None if the guard is unavailable."""
-    try:
-        return float(ncii_guard.classify(prompt)["p_ncii"])
-    except Exception as exc:  # pragma: no cover - demo must survive an outage
-        print(f"[guard] scoring failed ({exc!r}); allowing prompt")
-        return None
-
-
-def _screen_prompt(prompt: str, editing: bool) -> None:
-    """Reject NCII edit requests before a GPU worker is allocated."""
-    if not editing and not GUARD_SCREEN_TEXT_TO_IMAGE:
-        return
-    score = _guard_score(prompt)
-    if score is None:
-        return
-    if score >= GUARD_THRESHOLD:
-        print(f"[guard] blocked prompt (p_ncii={score:.3f})")
-        raise gr.Error(
-            "This prompt was flagged by a content filter and wasn't run."
-        )
-    if score >= 0.3:
-        print(f"[guard] allowed borderline prompt (p_ncii={score:.3f})")
 
 
 
@@ -468,8 +441,7 @@ def generate(
     img_cfg_scale = _clamp(img_cfg_scale, 1.0, 4.0, DEFAULT_IMG_CFG_SCALE)
 
     pil_images = _normalize_images(images)
-    # Screen on CPU, before any GPU worker is allocated.
-    _screen_prompt(prompt, editing=bool(pil_images))
+    # Screen on CPU, before any GPU worker is allocated
     budget = _input_pixel_budget(len(pil_images))
     pil_inputs = [_prep_input_image(img, budget) for img in pil_images]
     if pil_inputs: