File size: 483 Bytes
1ddcf83
 
 
 
 
56f6e37
1ddcf83
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
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 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"]