diff --git a/flake.nix b/flake.nix index 21853e1..c8947b4 100644 --- a/flake.nix +++ b/flake.nix @@ -52,7 +52,7 @@ inherit (self.packages."${prev.system}") objectifier yolo-cli; }; objectifier = final: prev: { - inherit (self.packages."${prev.system}") objectifier; + inherit (self.packages."${prev.system}") objectifier yolo-data; }; yolo-cli = final: prev: { inherit (self.packages."${prev.system}") yolo-cli; diff --git a/objectifier-module.nix b/objectifier-module.nix index 9c4fd92..1a21eea 100644 --- a/objectifier-module.nix +++ b/objectifier-module.nix @@ -1,3 +1,5 @@ + + { config, lib, pkgs, ... }: with lib; @@ -44,9 +46,9 @@ in { reloadIfChanged = true; path = with pkgs; [ pythonYolo ]; environment = { - OBJECTIFIER_YOLOV3_CONFIG = "${yolo-data}/yolov3.cfg"; - OBJECTIFIER_YOLOV3_WEIGHTS = "${yolo-data}/yolov3.weights"; - OBJECTIFIER_YOLOV3_LABELS = "${yolo-data}/labels"; + OBJECTIFIER_YOLOV3_CONFIG = "${pkgs.yolo-data}/yolov3.cfg"; + OBJECTIFIER_YOLOV3_WEIGHTS = "${pkgs.yolo-data}/yolov3.weights"; + OBJECTIFIER_YOLOV3_LABELS = "${pkgs.yolo-data}/labels"; OBJECTIFIER_BUFFER_SIZE = 524288; }; serviceConfig = {