CablePoleSegmentation / Dockerfile
blitzkrieg0000's picture
Update Dockerfile
dab22b2 verified
raw
history blame
405 Bytes
FROM python:3.11.5
RUN useradd -m -u 1000 user
WORKDIR /app
COPY --chown=user ./requirements.txt requirements.txt
RUN apt-get update && apt-get install ffmpeg libsm6 libxext6 -y
RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
RUN pip install gradio opencv-python pandas ultralytics
COPY --chown=user . /app
EXPOSE 1071
CMD ["python", "/app/UI/Main.py"]