FROM nvidia/cuda:12.1.0-runtime-ubuntu22.04 AS base
WORKDIR /app
COPY requirements.worker.txt .
RUN apt-get update && apt-get install -y python3-pip ffmpeg
RUN pip3 install -r requirements.worker.txt
COPY . .
CMD ["python3","main.py"]