update

Dockerfile

@@ -18,10 +18,9 @@ RUN conda run -n gradio pip install --upgrade pip
 # RUN conda install -n gradio -c conda-forge open-clip-torch pytorch-lightning
 # RUN conda run -n gradio pip install trimesh xatlas scikit-learn opencv-python omegaconf
 
-
 # Set the environment variable to use the gradio environment by default
-RUN echo "source activate gradio" > ~/.bashrc
-ENV PATH /opt/conda/envs/gradio/bin:$PATH
+# RUN echo "source activate gradio" > ~/.bashrc
+# ENV PATH /opt/conda/envs/gradio/bin:$PATH
 
 # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user
@@ -32,6 +31,10 @@ RUN conda create -n gradio-user python=3.11
 RUN conda run -n gradio-user pip install --upgrade pip
 RUN conda run -n gradio-user pip install diffusers["torch"] transformers accelerate xformers
 RUN conda run -n gradio-user pip install gradio 
+RUN conda run -n gradio pip install controlnet-aux
+RUN conda install -n gradio pytorch3d -c pytorch3d -c conda-forge
+RUN conda install -n gradio -c conda-forge open-clip-torch pytorch-lightning
+RUN conda run -n gradio pip install trimesh xatlas scikit-learn opencv-python omegaconf
 
 # Set home to the user's home directory
 ENV HOME=/home/user \

run.sh

@@ -4,13 +4,6 @@ CONDA_ENV=$(head -1 /code/environment.yml | cut -d" " -f2)
 eval "$(conda shell.bash hook)"
 conda activate gradio-user
 
-# Kill existing app.py process if running
-# if pgrep -f "python app.py" > /dev/null; then
-#     echo "Stopping existing app.py..."
-#     pkill -f "python app.py"
-#     sleep 2  # Small delay to ensure it stops
-# fi
-
 # Start app.py
 echo "Starting app.py..."
 python app.py