app.py

from __future__ import annotations

import os
from pathlib import Path

SPACE_ROOT = Path(__file__).resolve().parent
os.environ.setdefault("CHECKPOINT_PATH", str(SPACE_ROOT / "models/best.pt"))
os.environ.setdefault("OPENBLAS_NUM_THREADS", "1")
os.environ.setdefault("OMP_NUM_THREADS", "1")
os.environ.setdefault("MKL_NUM_THREADS", "1")
os.environ.setdefault("NUMEXPR_NUM_THREADS", "1")

from demo.app import CSS, build_demo  # noqa: E402


if __name__ == "__main__":
    build_demo().launch(css=CSS)