File size: 457 Bytes
29eecb6
 
 
 
 
 
 
 
 
 
 
 
b38dea1
29eecb6
 
 
374837f
 
 
0fe6487
29eecb6
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
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 chmod 777 /workspace
RUN mkdir -p /workspace/.cache/huggingface/hub
ENV TRANSFORMERS_CACHE=/workspace/.cache/huggingface/hub

CMD ["./start.sh"]