Forgot to copy yolo model in the last build stage

spacedriveapp · Dec 5, 2023 · 5ceff1a · 5ceff1a
1 parent d5b8c8c
commit 5ceff1a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -466,6 +466,8 @@ COPY --from=layer-99-protoc "${PREFIX}/licenses/." "${OUT}/licenses"
 COPY --from=layer-99-pdfium "${OUT}/." "$OUT"
 COPY --from=layer-99-pdfium "${PREFIX}/licenses/." "${OUT}/licenses"
 
+COPY --from=layer-99-yolo "${OUT}/." "$OUT"
+
 COPY --from=layer-99-onnx "${OUT}/." "$OUT"
 COPY --from=layer-99-onnx "${PREFIX}/srv/." "${OUT}/srv"
 COPY --from=layer-99-onnx "${PREFIX}/licenses/." "${OUT}/licenses"

diff --git a/stages/99-yolov8.sh b/stages/99-yolov8.sh
@@ -12,5 +12,6 @@ case "$TARGET" in
     ;;
 esac
 
-mkdir -p "${OUT}/models"
+mkdir -p "$OUT"/{models,licenses}
 curl -LSsO "${OUT}/models/yolov8s.onnx" "https://github.com/spacedriveapp/native-deps/releases/download/${_tag}/${_name}"
+curl -LSsO "https://raw.githubusercontent.com/ultralytics/assets/main/LICENSE" "${OUT}/licenses/yolov8s.LICENSE"