Skip to content

Commit

Permalink
fix image?
Browse files Browse the repository at this point in the history
  • Loading branch information
tomkosm committed Dec 26, 2024
1 parent fe3f67a commit 3cbd6ce
Show file tree
Hide file tree
Showing 4 changed files with 11 additions and 6 deletions.
2 changes: 1 addition & 1 deletion .github/workflows/image.yml
Original file line number Diff line number Diff line change
Expand Up @@ -28,5 +28,5 @@ jobs:

- name: 'Build Inventory Image'
run: |
docker build . --tag ghcr.io/mendableai/mineru-api:latest
docker build . --progress=plain --tag ghcr.io/mendableai/mineru-api:latest
docker push ghcr.io/mendableai/mineru-api:latest
5 changes: 4 additions & 1 deletion Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@ ENV PYTHONUNBUFFERED True

ENV DEBIAN_FRONTEND noninteractive
RUN apt-get update && apt-get install -y tzdata
ENV TZ Asia/Tokyo
# ENV TZ Asia/Tokyo

RUN apt-get update && \
apt-get install --yes --no-install-recommends curl g++ libopencv-dev python3.10 && \
Expand All @@ -27,4 +27,7 @@ COPY magic-pdf.gpu.json /root/magic-pdf.json

RUN python3.10 download_models.py

RUN python3.10 download_models2.py


CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "3000"]
5 changes: 1 addition & 4 deletions download_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,4 @@
"opendatalab/PDF-Extract-Kit-1.0",
local_dir="/root/PDF-Extract-Kit",
allow_patterns=mineru_patterns,
)
layoutreader_model_dir = snapshot_download(
"ppaanngggg/layoutreader", local_dir="/root/layoutreader"
)
)
5 changes: 5 additions & 0 deletions download_models2.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
from modelscope import snapshot_download

layoutreader_model_dir = snapshot_download(
"ppaanngggg/layoutreader", local_dir="/root/layoutreader"
)

0 comments on commit 3cbd6ce

Please sign in to comment.