prepare.sh

echo " PREPARING for HUGGING FACE "

if [ -d "./omnitool/" ]; then
    echo "[_] OMNITOOL ALREADY EXISTS "
else
    echo "[x] CLONING FOR OMNITOOL GIT "
    git clone https://github.com/omnitool-ai/omnitool
fi

if [ -d "./omnitool/.git" ]; then
    echo "[x] REMOVING GIT files "
    rm -rf ./omnitool/.git/
    rm ./omnitool/.gitignore
    rm ./omnitool/.gitattributes
else
    echo "[_] NO GIT FILES DETECTED "
fi

if [ -f "./omnitool/packages/omni-server/config.default/models/nsfwjs/mobilenet-v2-quant/group1-shard1of1" ]; then
    echo "[x] REMOVING large file from OMNITOOL distribution "
    rm ./omnitool/packages/omni-server/config.default/models/nsfwjs/mobilenet-v2-quant/group1-shard1of1
else
    echo "[_] NO LARGE FILE DETECTED "
fi

if [ "$(ls -A ./omnitool/assets)" ]; then
    echo "[x] REMOVING ASSETS "
    rm -rf ./omnitool/assets/
    mkdir -p ./omnitool/assets
else
    echo "[_] NO ASSETS DETECTED "
fi