Skip to content

Commit

Permalink
Fetch models from GitHub
Browse files Browse the repository at this point in the history
  • Loading branch information
frafra committed Apr 30, 2024
1 parent d657389 commit 2e37f7e
Showing 1 changed file with 10 additions and 1 deletion.
11 changes: 10 additions & 1 deletion Dockerfile
Original file line number Diff line number Diff line change
@@ -1,5 +1,15 @@
FROM python:3.10

ADD \
https://github.com/NINAnor/fisk-ai/releases/download/v0.1.0/v8m-classes-augmented.pt \
https://github.com/NINAnor/fisk-ai/releases/download/v0.1.0/v8s-640-classes-augmented-backgrounds.pt \
https://github.com/NINAnor/fisk-ai/releases/download/v0.1.0/yolov8l.pt \
https://github.com/NINAnor/fisk-ai/releases/download/v0.1.0/yolov8m.pt \
https://github.com/NINAnor/fisk-ai/releases/download/v0.1.0/yolov8n.pt \
https://github.com/NINAnor/fisk-ai/releases/download/v0.1.0/yolov8s.pt \
https://github.com/NINAnor/fisk-ai/releases/download/v0.1.0/yolov8x.pt \
/app/data/models/

# https://stackoverflow.com/questions/68036484/qt6-qt-qpa-plugin-could-not-load-the-qt-platform-plugin-xcb-in-even-thou#comment133288708_68058308
RUN --mount=type=cache,sharing=locked,target=/var/cache/apt \
--mount=type=cache,sharing=locked,target=/var/lib/apt \
Expand All @@ -20,5 +30,4 @@ RUN --mount=type=cache,sharing=locked,target=/root/.cache/pypoetry \
poetry install --no-root

COPY app app
COPY data/models/v8s-640-classes-augmented-backgrounds.pt data/models/
CMD ["poetry", "run", "python", "-c", "from app import main; main.main()"]

0 comments on commit 2e37f7e

Please sign in to comment.