|
|
|
FROM python:3.10-slim |
|
|
|
|
|
ENV PYTHONDONTWRITEBYTECODE=1 |
|
ENV PYTHONUNBUFFERED=1 |
|
ENV DEBIAN_FRONTEND=noninteractive |
|
|
|
|
|
RUN apt-get update && apt-get install -y \ |
|
build-essential \ |
|
libpq-dev \ |
|
&& rm -rf /var/lib/apt/lists/* |
|
|
|
|
|
WORKDIR /code |
|
|
|
|
|
COPY requirements.txt . |
|
RUN pip install --no-cache-dir -r requirements.txt |
|
|
|
|
|
RUN mkdir -p /root/.cache/huggingface |
|
VOLUME /root/.cache/huggingface |
|
|
|
|
|
RUN python -c "from transformers import AutoModel, AutoTokenizer; \ |
|
model_name='RufusRubin777/GOT-OCR2_0_CPU'; \ |
|
AutoModel.from_pretrained(model_name, local_files_only=False); \ |
|
AutoTokenizer.from_pretrained(model_name, local_files_only=False)" |
|
|
|
|
|
COPY . . |
|
|
|
|
|
EXPOSE 7863 |
|
|
|
|
|
CMD ["python", "app.py"] |
