platform/Dockerfile

FROM huggingface/transformers-pytorch-gpu:latest

ENV DEBIAN_FRONTEND=noninteractive

RUN apt-get update && apt-get install -y \
    git \
    wget \
    curl \
    vim \
    nano \
    build-essential \
    python3 \
    python3-pip

WORKDIR /app

COPY . .