| FROM pytorch/pytorch:2.1.0-cuda12.1-cudnn8-runtime | |
| # Set environment variables | |
| ENV PYTHONUNBUFFERED=1 | |
| # Install system dependencies | |
| RUN apt-get update && apt-get install -y --no-install-recommends \ | |
| git \ | |
| && rm -rf /var/lib/apt/lists/* | |
| # Set working directory | |
| WORKDIR /app | |
| # Install Python requirements | |
| RUN pip install --no-cache-dir transformers==4.37.2 gradio==3.38.0 pillow | |
| # Copy application files | |
| COPY simple_gpu_app.py /app/app.py | |
| # Make port 7860 available for the app | |
| EXPOSE 7860 | |
| # Start the application | |
| CMD ["python", "app.py"] |