FROM pytorch/pytorch:2.0.1-cuda11.7-runtime-ubuntu22.04

WORKDIR /app

RUN pip install --no-cache-dir \
    gradio \
    transformers \
    huggingface_hub

COPY app.py .

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