Spaces:
Sleeping
Sleeping
File size: 424 Bytes
0d17181 b8d478f 0d17181 be03b42 0d17181 be03b42 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | FROM python:3.10
ENV HF_HOME=/app/cache
WORKDIR /app
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt
# 👇 Pre-download model
RUN python -c "from transformers import AutoTokenizer, AutoModelForCausalLM; \
AutoTokenizer.from_pretrained('TinyLlama/TinyLlama-1.1B-Chat-v1.0'); \
AutoModelForCausalLM.from_pretrained('TinyLlama/TinyLlama-1.1B-Chat-v1.0')"
COPY app.py .
CMD ["python", "app.py"] |