| FROM ciwei123321/rose:latest |
|
|
| USER root |
| RUN apt-get update && apt-get install -y socat python3 python3-pip && \ |
| pip3 install huggingface-hub && \ |
| rm -rf /var/lib/apt/lists/* |
|
|
| ENV ROSE_ADMIN_PASSWORD=rose123 \ |
| TZ=Asia/Shanghai |
|
|
| RUN mkdir -p /app/data /app/strm |
|
|
| EXPOSE 7860 |
|
|
| CMD ["/bin/sh", "-c", "\ |
| if [ -n \"$HF_DATASET\" ] && [ -n \"$HF_TOKEN\" ]; then \ |
| echo '=== Syncing data from Hugging Face Hub ===' && \ |
| huggingface-cli download $HF_DATASET --repo-type=dataset --local-dir=/app/data --resume-download || true; \ |
| else \ |
| echo '=== No HF credentials, skipping data sync ==='; \ |
| fi; \ |
| echo '=== Starting Rose service (python3 run.pyc) ===' && \ |
| cd /app && \ |
| python3 run.pyc --port 8787 & \ |
| sleep 5 && \ |
| echo '=== Starting port forwarding 7860 -> 8787 ===' && \ |
| exec socat TCP-LISTEN:7860,fork,reuseaddr TCP:127.0.0.1:8787 \ |
| "] |