Dockerfile

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"]