Update Dockerfile

Dockerfile

@@ -3,6 +3,9 @@ FROM nvidia/cuda:11.3.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/*
@@ -47,6 +51,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 \
@@ -66,14 +71,14 @@ RUN cd /tmp && pip install -r requirements.txt
 RUN pip install https://download.is.tue.mpg.de/icon/HF/kaolin-0.11.0-cp38-cp38-linux_x86_64.whl
 RUN pip install https://download.is.tue.mpg.de/icon/HF/pytorch3d-0.7.0-cp38-cp38-linux_x86_64.whl
 
-# 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
 
+# Copy the current directory contents into the container at $HOME/app setting the owner to the user
+COPY --chown=user . $HOME/app
+
+# Set the working directory to the user's home directory
+WORKDIR $HOME/app
+
 CMD ["python", "app.py"]

README.md

@@ -1,11 +1,10 @@
 ---
 title: ICON - Clothed Human Digitization
-metaTitle: Making yourself an ICON, by Yuliang Xiu
+metaTitle: ICON-Avatarify from Photo
 emoji: 🤼
 colorFrom: indigo
 colorTo: yellow
 sdk: docker
-app_file: app.py
 pinned: true
 ---