pyment-public / docker /predict.Dockerfile
estenhl's picture
Finished docker container for preprocessing and prediction
eff52f6
raw
history blame contribute delete
571 Bytes
FROM python:3.10.2-slim
RUN mkdir -p /repos/pyment
COPY scripts /repos/pyment/scripts
COPY pyment /repos/pyment/pyment
COPY pyproject.toml /repos/pyment/
COPY README.md /repos/pyment/
COPY LICENSE.md /repos/pyment/
RUN pip install --upgrade pip poetry-core build && \
cd /repos/pyment && \
pip install --no-cache-dir .
RUN mkdir -p /.pyment/weights && \
chmod -R 1777 /.pyment
COPY checkpoints/pyment /.pyment/weights
CMD ["python", "/repos/pyment/scripts/predict_from_fastsurfer_folder.py", \
"/fastsurfer", \
"-d", "/output/predictions.csv"]