FROM python:3.9-slim

WORKDIR /app

RUN apt-get update && apt-get install -y wget

RUN --mount=type=secret,id=hf_token \
    wget https://huggingface.co/cigol123/YUGO-GPT-Q4_0-GGUF/resolve/main/yugogpt-q4_0.gguf \
    --header="Authorization: Bearer $(cat /run/secrets/hf_token)"

COPY requirements.txt .
RUN pip install -r requirements.txt

COPY . .

CMD ["python", "app.py"]