File size: 372 Bytes
29eecb6
 
 
 
 
 
 
 
 
 
 
 
b38dea1
29eecb6
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
FROM pytorch/pytorch:latest

RUN apt update && apt install -y && rm -rf /var/lib/apt/lists/*

COPY requirements.txt .
RUN python3 -m pip install -r requirements.txt

COPY setup.py VERSION .
COPY src ./src
RUN python3 -m pip install .

COPY "demo.py" .
EXPOSE 7860 8080
COPY start.sh .
RUN chmod +x start.sh

RUN mkdir /data
ENV TRANSFORMERS_CACHE=/data
CMD ["./start.sh"]