| FROM python:3.9-slim-bullseye | |
| RUN apt-get update | |
| RUN apt-get install git ffmpeg libsm6 libxext6 wget -y | |
| # Install dependencies | |
| COPY requirements.txt . | |
| RUN pip install -r requirements.txt | |
| RUN pip install gsutil | |
| # Create working directory | |
| RUN mkdir -p /usr/src/app | |
| WORKDIR /usr/src/app | |
| # Copy contents | |
| COPY . /usr/src/app | |
| ENTRYPOINT ["python3", "/usr/src/app/detect.py", "--weights", "weights/yolov5x_anime.pt", "--source", "/in", "--output", "/out"] |