FROM nvcr.io/nvidia/pytorch:24.06-py3

WORKDIR /app

COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt

# Fix: make models and cache dirs writable
RUN mkdir -p /app/models /app/.cache && chmod -R 777 /app/models /app/.cache

ENV HF_HOME=/app/.cache

COPY . .

EXPOSE 7860

CMD python server.py --model-repo SmilingWolf/wd-eva02-large-tagger-v3 --port 7860 --token $HF_TOKEN