FROM python:3.11-slim
RUN apt-get update && apt-get install -y libgl1 libglib2.0-0 && rm -rf /var/lib/apt/lists/*
WORKDIR /app
ENV YOLO_CONFIG_DIR=/tmp/Ultralytics MPLCONFIGDIR=/tmp/mpl HF_HOME=/tmp/hf
COPY requirements.txt .
RUN pip install --no-cache-dir torch torchvision --index-url https://download.pytorch.org/whl/cu121
RUN pip install --no-cache-dir -r requirements.txt
COPY . .
RUN python -c "from ultralytics import YOLO; YOLO('yolo11x.pt')"
EXPOSE 7860
CMD ["uvicorn","main:app","--host","0.0.0.0","--port","7860"]