Need to pass yolo-data to module

This commit is contained in:
niten 2023-01-06 17:31:58 -08:00
parent 7eef72712c
commit 31d032b5ef
2 changed files with 6 additions and 4 deletions

View File

@ -52,7 +52,7 @@
inherit (self.packages."${prev.system}") objectifier yolo-cli; inherit (self.packages."${prev.system}") objectifier yolo-cli;
}; };
objectifier = final: prev: { objectifier = final: prev: {
inherit (self.packages."${prev.system}") objectifier; inherit (self.packages."${prev.system}") objectifier yolo-data;
}; };
yolo-cli = final: prev: { yolo-cli = final: prev: {
inherit (self.packages."${prev.system}") yolo-cli; inherit (self.packages."${prev.system}") yolo-cli;

View File

@ -1,3 +1,5 @@
{ config, lib, pkgs, ... }: { config, lib, pkgs, ... }:
with lib; with lib;
@ -44,9 +46,9 @@ in {
reloadIfChanged = true; reloadIfChanged = true;
path = with pkgs; [ pythonYolo ]; path = with pkgs; [ pythonYolo ];
environment = { environment = {
OBJECTIFIER_YOLOV3_CONFIG = "${yolo-data}/yolov3.cfg"; OBJECTIFIER_YOLOV3_CONFIG = "${pkgs.yolo-data}/yolov3.cfg";
OBJECTIFIER_YOLOV3_WEIGHTS = "${yolo-data}/yolov3.weights"; OBJECTIFIER_YOLOV3_WEIGHTS = "${pkgs.yolo-data}/yolov3.weights";
OBJECTIFIER_YOLOV3_LABELS = "${yolo-data}/labels"; OBJECTIFIER_YOLOV3_LABELS = "${pkgs.yolo-data}/labels";
OBJECTIFIER_BUFFER_SIZE = 524288; OBJECTIFIER_BUFFER_SIZE = 524288;
}; };
serviceConfig = { serviceConfig = {