diff --git a/Dockerfile b/Dockerfile
index a92b9f45c226068a0ca15e3ae718d1a48150a67e..331f999161abe0b3b452a62fad8106bf3d65d48a 100755
--- a/Dockerfile
+++ b/Dockerfile
@@ -35,7 +35,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     torchvision \
     tqdm \
     pandas
-    
+
+
 # OpenSSH configuration
 RUN mkdir /var/run/sshd \
     && sed 's@session\s*required\s*pam_loginuid.so@session optional pam_loginuid.so@g' -i /etc/pam.d/sshd
@@ -52,8 +53,16 @@ RUN groupadd -g ${GID} ${USERNAME} \
     && mkdir -p ${HOME}/data ${HOME}/.vscode-server ${HOME}/.ssh\
     && chown ${USERNAME}:${USERNAME} ${HOME}/data ${HOME}/.vscode-server ${HOME}/.ssh
 
+# Copy your public key to the container
+COPY id_rsa.pub /home/${USERNAME}/.ssh/authorized_keys
+RUN chown ${USERNAME}:${USERNAME} /home/${USERNAME}/.ssh/authorized_keys \
+    && chmod 600 /home/${USERNAME}/.ssh/authorized_keys
+
+
 VOLUME [ "${HOME}/data", "${HOME}/.ssh", "${HOME}/.vscode-server"]
 
+
+
 # Expose SSH
 EXPOSE 22