FROM huggingface/transformers-pytorch-gpu:4.41.1

WORKDIR /app

COPY requirements.txt .
RUN pip install --upgrade pip && pip install -r requirements.txt

COPY . .

CMD ["python3", "app.py"]