@@
FROM pytorch/pytorch:2.6.0-cuda12.6-cudnn9-devel

#–––––––––––––––––––––––––––––––––––––––––––––––––––––––––––––––––
# 1.  Hugging Face cache lives in /data (.hf Space volume)       *
#–––––––––––––––––––––––––––––––––––––––––––––––––––––––––––––––––
ENV HF_HOME=/data/.huggingface

@@
 WORKDIR /workspace/spatial-dia
 ENV PYTHONUNBUFFERED=1

CMD ["/bin/bash"]

#–––––––––––––––––––––––––––––––––––––––––––––––––––––––––––––––––
# 2.  Boot script: pre-fetch weights once, then launch Gradio
#–––––––––––––––––––––––––––––––––––––––––––––––––––––––––––––––––
COPY entrypoint.sh /entrypoint.sh
RUN chmod +x /entrypoint.sh

CMD ["/entrypoint.sh"]