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"]