FROM huggingface/transformers-pytorch-gpu:latest

WORKDIR /repository

# Install diffusers from source (supports WanPipeline) + flask + imageio for video
RUN pip install --no-cache-dir 
    git+https://github.com/huggingface/diffusers.git 
    flask 
    accelerate 
    sentencepiece 
    protobuf 
    "imageio[ffmpeg]"

COPY server.py /repository/server.py
COPY handler.py /repository/handler.py
COPY requirements.txt /repository/requirements.txt

EXPOSE 8000

CMD ["python", "/repository/server.py"]