FROM python:3.9-slim

WORKDIR /app

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

ARG HF_TOKEN
ENV HF_TOKEN=${HF_TOKEN}
RUN wget https://huggingface.co/cigol123/YUGO-GPT-Q4_0-GGUF/resolve/main/yugogpt-q4_0.gguf --header="Authorization: Bearer ${HF_TOKEN}"

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

COPY . .

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