| FROM python:3.10-slim |
|
|
| ENV PYTHONDONTWRITEBYTECODE=1 |
| ENV PYTHONUNBUFFERED=1 |
| ENV DEBIAN_FRONTEND=noninteractive |
| ENV MODEL_PATH=RufusRubin777/GOT-OCR2_0_CPU |
|
|
| WORKDIR /app |
|
|
| |
| RUN apt-get update && apt-get install -y \ |
| build-essential \ |
| libpq-dev \ |
| && rm -rf /var/lib/apt/lists/* |
|
|
| |
| 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_modules.RufusRubin777.GOT_OCR2_0_CPU.modeling_GOT import GOTModel, GOTConfig; \ |
| from transformers import AutoTokenizer; \ |
| model_path='RufusRubin777/GOT-OCR2_0_CPU'; \ |
| config = GOTConfig.from_pretrained(model_path); \ |
| model = GOTModel.from_pretrained(model_path, config=config); \ |
| tokenizer = AutoTokenizer.from_pretrained(model_path)" |
|
|
| |
| COPY . . |
|
|
| EXPOSE 7863 |
|
|
| CMD ["python", "app.py"] |
