update requirements

bootstrap.py

@@ -0,0 +1,33 @@
+"""Runtime installation of the gigacheck package without its pinned deps.
+
+gigacheck pins ``torch==2.3.1`` in its ``install_requires``, which conflicts
+with the ``torch==2.8.0`` this demo needs. Installing gigacheck with
+``--no-deps`` lets pip skip that pin entirely; the inference path only needs
+``numpy``, ``scipy``, ``dacite`` and ``omegaconf`` on top of torch/transformers,
+which are declared in ``requirements.txt`` and installed normally.
+"""
+
+from __future__ import annotations
+
+import importlib.util
+import subprocess
+import sys
+
+from loguru import logger
+
+GIGACHECK_SPEC = "git+https://github.com/ai-forever/gigacheck"
+
+
+def ensure_gigacheck() -> None:
+    """Install the gigacheck package on first run, skipping its pinned deps.
+
+    No-op when gigacheck is already importable (e.g. installed in the image),
+    so repeated app restarts do not reinstall it.
+    """
+    if importlib.util.find_spec("gigacheck") is not None:
+        return
+    logger.info("Installing gigacheck with --no-deps from {}", GIGACHECK_SPEC)
+    subprocess.run(
+        [sys.executable, "-m", "pip", "install", "--no-deps", GIGACHECK_SPEC],
+        check=True,
+    )

models.py

@@ -6,9 +6,14 @@ import torch
 from loguru import logger
 from transformers import AutoModel
 
+from bootstrap import ensure_gigacheck
 from rendering import AiInterval
 from validation import CONFIG
 
+# gigacheck must be importable before any from_pretrained() loads the remote
+# model code, which does `from gigacheck.model... import ...`.
+ensure_gigacheck()
+
 try:
     import spaces
 
@@ -38,10 +43,10 @@ def select_device() -> str:
     decorated function, so the presence of the ``spaces`` package implies a GPU.
 
     Returns:
-        ``"cuda:0"`` when a GPU is (or will be) available, otherwise ``"cpu"``.
+        ``"cuda"`` when a GPU is (or will be) available, otherwise ``"cpu"``.
     """
     if torch.cuda.is_available() or HAS_SPACES:
-        return "cuda:0"
+        return "cuda"
     return "cpu"
 
 
@@ -65,7 +70,7 @@ detector_model = AutoModel.from_pretrained(
 )
 
 
-@gpu(duration=120)
+@gpu(duration=180)
 def classify_text(text: str) -> tuple[str, float, float]:
     """Classify a text as human-written or AI-generated.
 
@@ -85,7 +90,7 @@ def classify_text(text: str) -> tuple[str, float, float]:
     return label, p_human, p_ai
 
 
-@gpu(duration=120)
+@gpu(duration=180)
 def detect_intervals(text: str, conf_threshold: float) -> list[AiInterval]:
     """Detect character spans likely written by an AI model.
 

requirements.txt

@@ -6,4 +6,9 @@ sentencepiece
 loguru
 pydantic
 huggingface_hub
-git+https://github.com/ai-forever/gigacheck
+numpy
+scipy
+dacite
+omegaconf
+# gigacheck is installed at runtime with --no-deps (see bootstrap.py) to avoid
+# its torch==2.3.1 pin clashing with torch==2.8.0.