mike23415 commited on
Commit
7ada353
·
verified ·
1 Parent(s): c99b27e

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -5,7 +5,7 @@ WORKDIR /app
5
  COPY requirements.txt requirements.txt
6
  RUN pip install --no-cache-dir -r requirements.txt
7
 
8
- # Download model explicitly to avoid missing files
9
  RUN mkdir -p /app/cache && \
10
  HF_HOME=/app/cache python -c "from transformers import AutoModelForSeq2SeqLM; AutoModelForSeq2SeqLM.from_pretrained('t5-base')"
11
 
 
5
  COPY requirements.txt requirements.txt
6
  RUN pip install --no-cache-dir -r requirements.txt
7
 
8
+ # Create a cache directory and download the model
9
  RUN mkdir -p /app/cache && \
10
  HF_HOME=/app/cache python -c "from transformers import AutoModelForSeq2SeqLM; AutoModelForSeq2SeqLM.from_pretrained('t5-base')"
11