| # Base image with CUDA 12.8, cuDNN 9, PyTorch 2.7.0 | |
| FROM pytorch/pytorch:2.7.0-cuda12.8-cudnn9-devel | |
| # Set working directory | |
| ENV HOME=/usr/src/app | |
| WORKDIR $HOME | |
| # Install system dependencies, then clean up | |
| RUN apt-get update && \ | |
| apt-get install -y --no-install-recommends git wget && \ | |
| apt-get clean && \ | |
| rm -rf /var/lib/apt/lists/* | |
| # Copy application files | |
| COPY . . | |
| # Install Python dependencies | |
| RUN pip install --upgrade pip && \ | |
| pip install --no-cache-dir -r requirements.txt | |
| # Make /usr/src/app/ a writable directory | |
| RUN chmod -R 777 $HOME | |
| # Expose the Gradio port | |
| EXPOSE 7860 | |
| ENV GRADIO_SERVER_NAME="0.0.0.0" | |
| # Start the application | |
| CMD ["python", "app.py"] | |