Enzo Reis de Oliveira commited on
Commit
96362fc
·
1 Parent(s): 6f6fe4e

Adding Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +13 -0
Dockerfile ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM python:3.9.7
2
+
3
+ WORKDIR /app
4
+ COPY requirements.txt .
5
+ RUN pip install -r --no-cache-dir requirements.txt
6
+ # preload models
7
+ RUN python -c '\
8
+ from transformers import BartForConditionalGeneration, AutoTokenizer;\
9
+ AutoTokenizer.from_pretrained("ibm/materials.selfies-ted");\
10
+ BartForConditionalGeneration.from_pretrained("ibm/materials.selfies-ted")'
11
+ COPY . .
12
+
13
+ CMD ["python", "app.py"]