Update Dockerfile

Dockerfile

@@ -23,8 +23,19 @@ 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/[email protected]
 
+# Install aria2 directly using wget
+RUN apt-get update && apt-get install -y 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
+
 # Download checkpoint files using aria2
-RUN apt-get update && apt-get install -y 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
 RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/lxq007/DiffBIR/resolve/main/general_swinir_v1.ckpt -d $HOME/app/models -o general_swinir_v1.ckpt