Spaces:
Sleeping
Sleeping
| 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"] |