Update Dockerfile

Dockerfile

@@ -27,9 +27,10 @@ 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/open_clip@v2.20.0
 
+USER root
 # Update package lists and install aria2 without sudo
-RUN apt-get update && apt-get install -y --no-install-recommends aria2
-
+RUN apt-get update && apt-get install -y aria2
+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