Adding vim and other improvements

.gitignore

@@ -0,0 +1 @@
+.idea

Dockerfile

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

on_startup.sh

@@ -2,4 +2,19 @@
 # Write some commands here that will run on root user before startup.
 # For example, to clone transformers and install it in dev mode:
 # git clone https://github.com/huggingface/transformers.git
-# cd transformers && pip install -e ".[dev]"
+# cd transformers && pip install -e ".[dev]"
+git config --global credential.helper store
+
+## Clone the repository into a temporary directory
+#git clone https://huggingface.co/derek-thomas/tgi-benchmark-notebooks /tmp/tgi-benchmark-notebooks
+#
+## Move the contents to $HOME/app/notebooks
+#mv /tmp/tgi-benchmark-notebooks/0*.ipynb $HOME/app/notebooks
+#
+## Remove the temporary clone directory
+#rm -rf /tmp/tgi-benchmark-notebooks
+
+# 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="/data"
+NOTEBOOK_DIR="$HOME/app/notebooks/"
 
 jupyter-lab \
     --ip 0.0.0.0 \