Update Dockerfile

Dockerfile

@@ -23,8 +23,21 @@ RUN pip install -q einops pytorch_lightning gradio omegaconf xformers==0.0.20 tr
 # Install open_clip from GitHub
 RUN pip install -q git+https://github.com/mlfoundations/open_clip@v2.20.0
 
-# Install aria2
-RUN apt-get update && apt-get install -y aria2
+# Use sudo to install aria2 with elevated privileges
+USER root
+RUN apt-get update && apt-get install -y sudo wget && \
+    wget https://github.com/aria2/aria2/releases/download/release-1.36.0/aria2-1.36.0.tar.gz && \
+    tar -xzvf aria2-1.36.0.tar.gz && \
+    cd aria2-1.36.0 && \
+    ./configure && \
+    make && \
+    make install && \
+    cd .. && \
+    rm -rf aria2-1.36.0 && \
+    rm aria2-1.36.0.tar.gz
+
+# Switch back to the "user" user
+USER user
 
 # Download checkpoint files using aria2
 RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/lxq007/DiffBIR/resolve/main/general_full_v1.ckpt -d $HOME/app/models -o general_full_v1.ckpt