changed notebook location

Dockerfile

@@ -80,8 +80,9 @@ RUN --mount=target=/root/packages.txt,source=packages.txt \
 
 # RUN --mount=target=/root/on_startup.sh,source=on_startup.sh,readwrite \
 # 	bash /root/on_startup.sh
+ARG NOTEBOOK_DIR
 
-RUN mkdir /data && chown user:user /data
+RUN mkdir $NOTEBOOK_DIR && chown user:user $NOTEBOOK_DIR
 
 # sudo permissions
 RUN chown root:root /usr/bin/sudo && chmod 777 /usr/bin/sudo

backup.py

@@ -3,11 +3,11 @@ load_dotenv()
 from datasets import load_dataset
 from huggingface_hub import CommitScheduler
 from huggingface_hub import HfApi
-
+import os
 
 
 dataset_name = "Anuj-Panthri/JupyterLab_Storage"
-work_dir = "/home/user/jupyter"
+work_dir = os.environ['NOTEBOOK_DIR']
 dataset_save_path = "."
 
 def push_to_hub():

restore.py

@@ -1,8 +1,9 @@
 from huggingface_hub import snapshot_download
 from pathlib import Path
+import os
 
 dataset_name = "Anuj-Panthri/JupyterLab_Storage"
-work_dir = "/home/user/jupyter"
+work_dir = os.environ['NOTEBOOK_DIR']
 dataset_save_path = "."
 
 snapshot_download(

start_server.sh

@@ -3,8 +3,6 @@ JUPYTER_TOKEN="${JUPYTER_TOKEN:=huggingface}"
 
 echo "Starting Jupyter Lab with token $JUPYTER_TOKEN"
 
-NOTEBOOK_DIR="/home/user/jupyter"
-
 jupyter-lab \
     --ip 0.0.0.0 \
     --port 7860 \
@@ -16,4 +14,4 @@ jupyter-lab \
     --ServerApp.disable_check_xsrf=True \
     --LabApp.news_url=None \
     --LabApp.check_for_updates_class="jupyterlab.NeverCheckForUpdate" \
-    --notebook-dir=$NOTEBOOK_DIR
+    --notebook-dir=${NOTEBOOK_DIR}