Spaces:
Running
Running
File size: 419 Bytes
d1a4739 2530cc1 d1a4739 5dbff86 d1a4739 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
FROM python:3.9.7
WORKDIR /app
COPY requirements.txt .
RUN pip install -r --no-cache-dir requirements.txt
# preload models
RUN pip install torch_scatter==2.1.2
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"] |