Add local model asset catalog

2026-07-06 20:59:03 +02:00
parent 9e20cc82ae
commit 6e2a8cbdc4
33 changed files with 660 additions and 7 deletions
@@ -9,7 +9,7 @@ from typing import Iterable


 SUPPORTED_MODEL_SUFFIXES = {".pt", ".onnx", ".engine"}
-ENV_KEYS = ("GEOINTEL_INSTALL_AI", "YOLO_ENABLED", "YOLO_MODEL_PATH")
+ENV_KEYS = ("GEOINTEL_INSTALL_AI", "YOLO_ENABLED", "YOLO_MODELS_DIR", "YOLO_MODEL_PATH")


 def _candidate_paths(models_dir: Path) -> list[Path]:
@@ -149,6 +149,7 @@ def configure(args: argparse.Namespace) -> tuple[int, dict[str, object]]:
    updates = {
        "GEOINTEL_INSTALL_AI": "true",
        "YOLO_ENABLED": "true",
+        "YOLO_MODELS_DIR": args.container_model_dir.rstrip("/"),
        "YOLO_MODEL_PATH": selected_container_path,
    }
    payload.update(