| # Use an official PyTorch image with CUDA support as a base image | |
| FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime | |
| # Install additional Python dependencies | |
| RUN pip install transformers==4.31.0 gradio==3.40.1 | |
| # Expose the default port for Gradio | |
| EXPOSE 7860 | |
| # Copy your app files | |
| COPY . /app | |
| WORKDIR /app | |
| # Run the Gradio app | |
| CMD |