FROM python:3.10

RUN apt-get update && apt-get install -y \
    ffmpeg \
    libsm6 \
    libxext6 

WORKDIR /app

COPY . /app/

RUN pip install --upgrade pip
RUN pip install torch
RUN pip install -r requirements.txt
RUN pip install git+https://github.com/sajjjadayobi/FaceLib.git
RUN mkdir checkpoints
RUN wget -O ./checkpoints/inswapper_128.onnx https://huggingface.co/ashleykleynhans/inswapper/resolve/main/inswapper_128.onnx

CMD ["python3", "app.py", "-H", "0.0.0.0", "-p", "7860"]