File size: 251 Bytes
d7c7a6c
 
 
3ccf041
d7c7a6c
3ccf041
d7c7a6c
3ccf041
81ef716
d7c7a6c
 
3ccf041
81ef716
1
2
3
4
5
6
7
8
9
10
11
12
13
14
FROM huggingface/transformers-pytorch-gpu:latest

RUN pip install --upgrade pip
RUN pip install transformers torch fastapi uvicorn

COPY app.py /app/app.py

WORKDIR /app

EXPOSE 8080

CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8080"]