Moving to data and adding notebooks

Dockerfile

@@ -60,8 +60,8 @@ RUN curl -sLo ~/miniconda.sh https://repo.continuum.io/miniconda/Miniconda3-py31
 WORKDIR $HOME/app
 
 # Create folder for notebooks
-RUN mkdir -p $HOME/app/notebooks && \
-    chown user:user $HOME/app/notebooks
+RUN mkdir -p /data/notebooks && \
+    chown user:user /data/notebooks
 
 #######################################
 # Start root user section

on_startup_user.sh

@@ -14,13 +14,21 @@
 #rm -rf /tmp/tgi-benchmark-notebooks
 
 # Install llmperf
-cd ~/app
+cd /data
 git clone https://github.com/ray-project/llmperf.git
 cd llmperf
 git checkout afd137a
-mv ~/app/setup.py .
 pip install -e .
 
+# Get notebooks
+git clone https://huggingface.co/derek-thomas/tgi-notebooks-optimization /tmp/tgi-notebooks-optimization
+
+# Move the contents to /data
+mv /tmp/tgi-notebooks-optimization/0*.ipynb /data/notebooks
+
+# Remove the temporary clone directory
+rm -rf /tmp/tgi-notebooks-optimization
+
 # Add dark theme
 mkdir -p ~/.jupyter/lab/user-settings/@jupyterlab/apputils-extension/ && \
     echo '{ "theme":"JupyterLab Dark" }' >  ~/.jupyter/lab/user-settings/@jupyterlab/apputils-extension/themes.jupyterlab-settings

start_server.sh

@@ -3,7 +3,7 @@ JUPYTER_TOKEN="${JUPYTER_TOKEN:=huggingface}"
 
 echo "Starting Jupyter Lab with token $JUPYTER_TOKEN"
 
-NOTEBOOK_DIR="$HOME/app/"
+NOTEBOOK_DIR="/data/"
 
 jupyter-lab \
     --ip 0.0.0.0 \