Spaces:
Running
Running
File size: 382 Bytes
d1a4739 2530cc1 d1a4739 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
FROM python:3.9.7
WORKDIR /app
COPY requirements.txt .
RUN pip install -r --no-cache-dir requirements.txt
# preload models
RUN python -c '\
from transformers import BartForConditionalGeneration, AutoTokenizer;\
AutoTokenizer.from_pretrained("ibm/materials.selfies-ted");\
BartForConditionalGeneration.from_pretrained("ibm/materials.selfies-ted")'
COPY . .
CMD ["python", "app.py"] |