H-Liu1997 commited on
Commit
da265c6
·
1 Parent(s): e153101

fix: remove pre-download step, model downloads at runtime startup

Browse files
Files changed (1) hide show
  1. Dockerfile +0 -3
Dockerfile CHANGED
@@ -20,8 +20,5 @@ ENV HOME=/home/user \
20
  WORKDIR /home/user/app
21
  COPY --chown=user:user . .
22
 
23
- # Pre-download model during build (avoids slow cold start)
24
- RUN python3 -c "from transformers import AutoModel; AutoModel.from_pretrained('ShandaAI/FloodDiffusionTiny', trust_remote_code=True)"
25
-
26
  EXPOSE 7860
27
  CMD ["python3", "app.py", "--port", "7860"]
 
20
  WORKDIR /home/user/app
21
  COPY --chown=user:user . .
22
 
 
 
 
23
  EXPOSE 7860
24
  CMD ["python3", "app.py", "--port", "7860"]