| FROM pytorch/pytorch:2.4.0-cuda12.1-cudnn9-runtime | |
| WORKDIR /app | |
| COPY requirements.txt . | |
| RUN pip install --no-cache-dir -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu121 | |
| COPY app.py . | |
| ENV PORT=7860 | |
| EXPOSE 7860 | |
| CMD ["python", "app.py"] | |