Update Dockerfile

Dockerfile

@@ -3,6 +3,9 @@ FROM nvidia/cuda:11.6.1-cudnn8-devel-ubuntu20.04
 ARG DEBIAN_FRONTEND=noninteractive
 
 RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
+    curl \
+    git \
+    wget \
     libgl1 \ 
     freeglut3-dev \
     unzip \
@@ -20,6 +23,7 @@ RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-ins
     libeigen3-dev \
     python3.8 \
     python3-pip \
+    python-is-python3 \
     nvidia-cuda-toolkit \
     opencv-python-headless  \
     && rm -rf /var/lib/apt/lists/*
@@ -46,6 +50,7 @@ ENV LIBRARY_PATH=${CUDA_HOME}/lib64/stubs:${LIBRARY_PATH}
 
 # Set home to the user's home directory
 ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH \
     PYTHONPATH=$HOME/app \
 	PYTHONUNBUFFERED=1 \
 	GRADIO_ALLOW_FLAGGING=never \
@@ -63,14 +68,14 @@ RUN cd /tmp && pip install -r requirements.txt
 RUN pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py38_cu116_pyt1130/download.html
 RUN python -c "import torch; print(torch.version.cuda)"
 
-# Set the working directory to the user's home directory
-WORKDIR $HOME
-
-# Copy the current directory contents into the container at $HOME/app setting the owner to the user
-COPY --chown=user . $HOME
 RUN chmod 777 $HOME
-
 ENV TRANSFORMERS_CACHE=/tmp
 ENV MPLCONFIGDIR=/tmp
 
+# Set the working directory to the user's home directory
+WORKDIR $HOME/app
+
+# Copy the current directory contents into the container at $HOME/app setting the owner to the user
+COPY --chown=user . $HOME/app
+
 CMD ["python", "app.py"]

README.md

@@ -1,11 +1,10 @@
 ---
-title: Unconstrained & Detailed Clothed Human Digitization (ECON + ControlNet)
+title: ECON - Clothed Human Digitization
 metaTitle: ECON-Avatarify from Photo
 emoji: 🤼
 colorFrom: green
 colorTo: pink
 sdk: docker
-app_file: app.py
 pinned: true
 ---