Update Dockerfile

Dockerfile

@@ -5,7 +5,7 @@ ENV DEBIAN_FRONTEND=noninteractive
 # Set the MKL_THREADING_LAYER environment variable to GNU
 ENV MKL_THREADING_LAYER=GNU
 
-RUN apt-get update && apt-get install -y git wget libgl1-mesa-glx libglib2.0-0 ffmpeg libx264-dev build-essential cmake
+RUN apt-get update && apt-get install -y git wget libgl1-mesa-glx libglib2.0-0 ffmpeg libx264-dev build-essential cmake aria2
 
 RUN useradd -m -u 1000 user
 
@@ -27,7 +27,7 @@ WORKDIR $HOME/app
 
 RUN git clone -b dev https://github.com/camenduru/DragNUWA $HOME/app 
     
-RUN wget https://huggingface.co/camenduru/DragNUWA/resolve/main/4799.pth -O $HOME/app/models/4799.pth
+RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/camenduru/DragNUWA/resolve/main/4799.pth -d $HOME/app/models -o 4799.pth
     
 # Install dependencies
 RUN pip install --no-cache-dir -r environment.txt