Update Dockerfile

Dockerfile

@@ -1,49 +1,13 @@
-# Use the base image with CUDA and Ubuntu
-FROM nvidia/cuda:12.0.1-devel-ubuntu20.04
+FROM pytorch/pytorch:2.1.2-cuda11.8-cudnn8-devel
 
-RUN apt-get update && apt-get install -y git wget curl
-ENV DEBIAN_FRONTEND=noninteractive
-RUN ln -snf /usr/share/zoneinfo/Asia/Kolkata /etc/localtime && echo "Asia/Kolkata" > /etc/timezone
+# Copy all files from the current directory to /root/roop-unleashed
+COPY . /root/roop-unleashed
 
-# Set the working directory in the container
-WORKDIR /app
+# Set the working directory
+WORKDIR /root/roop-unleashed
 
-# Install necessary dependencies for adding repositories
-RUN apt-get update && \
-    apt-get install -y software-properties-common
+# Run the setup.sh script
+RUN chmod +x setup.sh && ./setup.sh
 
-# Add the deadsnakes PPA for Python 3.11
-RUN add-apt-repository ppa:deadsnakes/ppa && \
-    apt-get update
-
-# Install Python 3.11 and pip
-# Install Python 3.11 and development packages
-RUN apt-get install -y python3.11 python3.11-venv python3.11-dev libgl1-mesa-glx ffmpeg
-
-# Download and install the latest pip using get-pip.py
-RUN wget https://bootstrap.pypa.io/get-pip.py && \
-    python3.11 get-pip.py && \
-    rm get-pip.py
-
-    
-# Set the default Python version to 3.11
-RUN update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.11 1
-
-RUN ln -sf /usr/bin/python3.11 /usr/bin/python && \
-    ln -sf /usr/bin/python3.11 /usr/bin/python3
-    
-RUN python --version
-RUN python3 --version
-RUN pip --version
-RUN pip3 --version
-
-WORKDIR /workspace
-RUN git clone https://github.com/C0untFloyd/roop-unleashed
-WORKDIR /workspace/roop-unleashed
-RUN pip3 install -r requirements.txt
-
-# Expose port 7860
-EXPOSE 7860
-
-# Set the entrypoint to run the application
-CMD ["python", "run.py"]
+# Command to run JupyterLab and the Python script on startup
+CMD jupyter lab --ip='*' --NotebookApp.token='' --NotebookApp.password='' & python run.py