Update Dockerfile

Dockerfile

@@ -27,21 +27,8 @@ RUN pip install -q einops pytorch_lightning gradio omegaconf xformers==0.0.16 tr
 # Install open_clip from GitHub
 RUN pip install -q git+https://github.com/mlfoundations/[email protected]
 
-# Use sudo to install aria2 with elevated privileges
-USER root
-RUN apt-get update && apt-get install -y sudo wget libgl1-mesa-glx && \
-    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
+# Install aria2 for downloading files
+RUN apt-get update && apt-get install -y aria2
 
 # 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