update docker

Dockerfile

@@ -18,7 +18,9 @@ RUN wget -O - https://www.openssl.org/source/openssl-1.1.1u.tar.gz | tar zxf - \
 ENV SSL_CERT_DIR=/usr/lib/ssl/certs
 ENV LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH
 
-RUN mkdir -p /home/user/.cache/torch && chmod -R u+rwx /home/user/.cache/torch
+ENV TORCH_HOME /home/user/.cache/torch
+# RUN mkdir -p /home/user/.cache/torch && chmod -R u+rwx /home/user/.cache/torch
+RUN mkdir -p /home/user/.cache/huggingface/datasets && chmod -R u+rwx /home/user/.cache/huggingface/datasets
 
 RUN apt-get update
 RUN apt-get install build-essential libssl-dev ca-certificates libasound2 wget