| FROM python:3.11-slim |
|
|
| RUN apt-get update && apt-get install -y --no-install-recommends \ |
| build-essential git curl libgl1 \ |
| && rm -rf /var/lib/apt/lists/* |
|
|
| RUN pip install --upgrade pip \ |
| && pip install \ |
| fastapi==0.115.0 uvicorn[standard]==0.30.6 \ |
| pillow==10.4.0 pydantic==2.9.2 sqlalchemy==2.0.35 \ |
| python-multipart==0.0.9 \ |
| && pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 |
|
|
| RUN pip install \ |
| hydra-core \ |
| iopath \ |
| matplotlib \ |
| opencv-python==4.8.0.76 \ |
| numpy==1.26.4 \ |
| transformers \ |
| sentence-transformers \ |
| pydantic[email] |
|
|
| WORKDIR /app |
|
|
| COPY . /app |
| EXPOSE 8000 |
| CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"] |
|
|
