Update Dockerfile

Dockerfile

@@ -1,19 +1,22 @@
 FROM cnstark/pytorch:2.0.1-py3.10.11-ubuntu22.04
 
-WORKDIR /code
+RUN apt-get install -y sudo
+RUN useradd -rm -d /home/hf -s /bin/bash -g root -G sudo -u 1001 hf
+USER hf
+
+WORKDIR /home/hf
 
-RUN apt-get update && apt-get install -y git
+RUN sudo apt-get update && sudo apt-get install -y git
 #RUN pip3 install torch==2.0.1 --index-url https://download.pytorch.org/whl/cpu
 #RUN git clone https://github.com/rupeshs/fastsdcpu && chmod a+x ./fastsdcpu/install.sh && chmod a+x ./fastsdcpu/start-webui.sh && /code/fastsdcpu/install.sh
 
 RUN git clone https://github.com/rupeshs/fastsdcpu
 
-WORKDIR /code/fastsdcpu
+WORKDIR /home/hf/fastsdcpu
 RUN chmod a+x install.sh && chmod a+x start-webui.sh 
-RUN /code/fastsdcpu/install.sh || { exit 0; }
+RUN /home/hf/fastsdcpu/install.sh || { exit 0; }
 
-RUN apt-get install -y sudo
-RUN useradd -rm -d /home/hf -s /bin/bash -g root -G sudo -u 1001 hf
-USER hf
+ENV TRANSFORMERS_CACHE /home/hf/.cache
+ENV HF_HOME /home/hf/.cache
 
 CMD ["./start-webui.sh"]