#!/bin/bash
# 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]"
git config --global credential.helper store

# Create the target directory if it doesn't already exist
mkdir -p $HOME/app/notebooks

# 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

echo "All right on the HF site"