omnitool_init.sh

#!/bin/bash
echo "[->] START "

echo "[->] CHECKING EXISTING /DATA "
if [ -d "/data" ]; then
    echo "$(ls -l /data)" 
    echo "[<] creating symlink"
    ln -s /data /app/omnitool/packages/omni-server/data.local
else
    echo "[<] NO persistent /DATA DETECTED. You can add 20Gig of persistent storage as a paid option to Hugging Face"
fi


#         rm -rf ./omnitool/packages/omni-server/data.local
#     fi

#     if [ -d "./omnitool/packages/omni-server/data.local" ]; then
#         echo "[v] DATA.LOCAL already present "
#         rm -rf ./omnitool/packages/omni-server/data.local         # hopefully it got copied over through the symlink in the previous run
#     fi

#     echo "[v] Creating the symlink "
#     ln -s /data ./omnitool/packages/omni-server/data.local
#     chmod 0777 ./omnitool/packages/omni-server/data.local
#     chown node ./omnitool/packages/omni-server/data.local
# else
#     echo "[v] NO persistent /DATA DETECTED. You can add 20Gig of persistent storage as a paid option to Hugging Face"
# fi

echo "[->] UPDATE OMNITOOL "
cd ./omnitool
output=$(git pull)
if echo "$output" | grep -q "Already up to date."; then
  echo "The repository is already up to date."
else
  echo "New data was fetched."
  echo "[->] YARN INSTALL"
  yarn
fi

cd ..
echo "[->] Updating permissions"
chmod -R 0777 .
chown -Rh node:node .

echo "[->] YARN START "
yarn start -u -rb -R blocks