rose / Dockerfile
wd21's picture
Update Dockerfile
910e52f verified
raw
history blame contribute delete
902 Bytes
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 \
"]