# Use the official Python image as the base image
FROM nvidia/cuda:11.0.3-base-ubuntu20.04


# Run update
RUN apt update


#Install Python

RUN apt-get install -y python3 python3-pip


# Set the working directory inside the container
WORKDIR /app


# Copy the contents of the cloned GitHub repository to the working directory in the container
COPY ./ /app

# Install Python dependencies
RUN pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118 xformers
RUN pip3 install -r requirements.txt


# Expose the port the application will be running on
EXPOSE 8188

# Set the environment variable for GPU support
ENV NVIDIA_VISIBLE_DEVICES all

# Run the Python program when the container starts
CMD ["python3", "main.py", "--listen","0.0.0.0"]