| FROM python:3.10-slim | |
| # Install g++ | |
| RUN apt-get update && apt-get install -y g++ | |
| WORKDIR /app | |
| COPY ./ /app | |
| # For GPU support: | |
| RUN pip3 install --extra-index-url https://download.pytorch.org/whl/cu118 torch torchvision | |
| RUN pip install -r requirements.txt | |
| CMD fastapi run --reload --host=0.0.0.0 --port=7860 | |