FROM python:3.11-slim

WORKDIR /app

RUN pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cpu && \
    pip install --no-cache-dir numpy

COPY src/ src/
COPY checkpoints/ checkpoints/
COPY experiments/ experiments/

EXPOSE 7860
ENV DEVICE=cpu
CMD ["python", "experiments/demo_server.py"]