conda create -n cleanup python=3.10 -y
conda activate cleanup
# conda install pytorch==1.9.0 torchvision==0.10.0 cudatoolkit=11.1 -c pytorch -c nvidia
conda install pytorch torchvision -c pytorch -y

pip install -r requirements.txt


# Check if lama directory exists
if [ ! -d "lama" ]; then
    # Clone dependency repos
    git clone https://github.com/advimman/lama.git
else
    echo "lama directory already exists. Skipping clone."
fi

# Check if big-lama.zip or big-lama directory exists
if [ ! -f "big-lama.zip" ] && [ ! -d "big-lama" ]; then
    # Download big-lama model
    curl -LJO https://huggingface.co/smartywu/big-lama/resolve/main/big-lama.zip
    unzip big-lama.zip
else
    echo "big-lama model already exists. Skipping download and extraction."
fi