Update Dockerfile

Dockerfile

@@ -23,17 +23,8 @@ 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 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
+# Install aria2
+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