| 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"] | |