File size: 316 Bytes
2f06744 589ab6d 2f06744 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | 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"]
|