| FROM python:3.9.7 | |
| WORKDIR /app | |
| COPY requirements.txt . | |
| RUN pip install -r 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"] |