diff --git a/Dockerfile b/Dockerfile index f819031..9488d64 100644 --- a/Dockerfile +++ b/Dockerfile @@ -3,6 +3,8 @@ FROM python:3.11-slim RUN apt-get update && apt-get install -y --no-install-recommends \ poppler-utils \ libgomp1 \ + libgl1 \ + libglib2.0-0t64 \ && rm -rf /var/lib/apt/lists/* WORKDIR /app diff --git a/api/batch_translate.py b/api/batch_translate.py index 0007fda..5efeda2 100644 --- a/api/batch_translate.py +++ b/api/batch_translate.py @@ -52,7 +52,7 @@ for arxiv_id in papers: translate( [pdf_path], output=TRANSLATED_DIR, lang_in='en', lang_out='zh', - service='deepseek', thread=4, model=model, + service='deepseek', model=model, ) mono = os.path.join(TRANSLATED_DIR, f"{arxiv_id}-mono.pdf") dual = os.path.join(TRANSLATED_DIR, f"{arxiv_id}-dual.pdf") diff --git a/models/doclayout_yolo_docstructbench_imgsz1024.onnx b/models/doclayout_yolo_docstructbench_imgsz1024.onnx new file mode 100644 index 0000000..6605cac Binary files /dev/null and b/models/doclayout_yolo_docstructbench_imgsz1024.onnx differ