gpu?

Dockerfile

@@ -9,7 +9,7 @@ RUN apt-get update && apt-get install -y tzdata
 # ENV TZ Asia/Tokyo
 
 RUN apt-get update && \
-    apt-get install --yes --no-install-recommends curl g++ libopencv-dev python3.10 && \
+    apt-get install --yes --no-install-recommends curl g++ libopencv-dev python3.10 && python3-pip \
     rm -rf /var/lib/apt/lists/*
 RUN curl -sSL https://install.python-poetry.org | POETRY_VERSION=${POETRY_VERSION} python3.10 -
 
@@ -24,12 +24,13 @@ RUN poetry config virtualenvs.create false && \
     rm -rf /root/.cache/pypoetry && \
     rm -rf /root/.cache/pip
 
+#use paddlegpu
+RUN pip install paddlepaddle-gpu==3.0.0b1 -i https://www.paddlepaddle.org.cn/packages/stable/cu124/
+
 COPY . ./
 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"]

download_models.py

@@ -12,4 +12,8 @@
     "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"
+)