| FROM python:3.11-slim | |
| ENV PYTHONUNBUFFERED=1 \ | |
| OMP_NUM_THREADS=2 \ | |
| NTHREADS=2 \ | |
| HF_HOME=/tmp/hf \ | |
| HF_HUB_OFFLINE=1 \ | |
| TRANSFORMERS_OFFLINE=1 \ | |
| TOKENIZERS_PARALLELISM=false | |
| WORKDIR /app | |
| # lightgbm needs this system library | |
| RUN apt-get update && apt-get install -y --no-install-recommends libgomp1 \ | |
| && rm -rf /var/lib/apt/lists/* | |
| # --- ESM block: DELETE these 2 lines (and the 'transformers' line in requirements.txt) for Stage 1 --- | |
| RUN pip install --no-cache-dir torch==2.4.1 --index-url https://download.pytorch.org/whl/cpu | |
| # --- end ESM block --- | |
| COPY requirements.txt . | |
| RUN pip install --no-cache-dir -r requirements.txt | |
| COPY features.py predict.py ./ | |
| COPY weights ./weights | |