| FROM pytorch/pytorch:2.4.1-cuda12.4-cudnn9-runtime | |
| ARG DEBIAN_FRONTEND=noninteractive | |
| ENV PYTHONUNBUFFERED=1 | |
| RUN useradd -m -u 1000 user | |
| RUN apt-get update && apt-get install build-essential ffmpeg libsm6 libxext6 git git-lfs --no-install-recommends -y \ | |
| && apt-get clean \ | |
| && rm -rf /var/lib/apt/lists/* | |
| RUN pip install --no-cache-dir gradio opencv-python pandas ultralytics onnx onnxruntime | |
| USER user | |
| WORKDIR /app | |
| COPY --chown=user ./ /app | |
| EXPOSE 1071 | |
| CMD ["python", "/app/UI/Main.py"] |