# read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
# you will also find guides on how best to write your Dockerfile

# Include base image
FROM docker.io/pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime

# Define working directory
WORKDIR /workspace/

# Set timezone
ENV TZ=Asia/Tokyo
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone

# Install dependencies
RUN apt-get update && apt-get -y install libgl1 libglib2.0-0 vim
RUN apt-get autoremove -y && apt-get clean -y

# Add pretrained model
ADD seg2art ./seg2art
ADD static ./static
ADD templates ./templates
ADD utils ./utils

# Add necessary files
ADD app.py ./

# pip install
ADD requirements.txt ./
RUN pip install -r requirements.txt

# Run server
CMD [ "python", "-u", "./app.py" ]