Spaces:
Sleeping
Sleeping
Update Dockerfile
Browse files- 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 |
-
#
|
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 |
|