Create prepare.sh

sd-webui/prepare.sh

@@ -0,0 +1,64 @@
+#!/bin/bash  
+  
+if [ ! -f "${ROOT}/prepare-v1.ready" ]; then
+
+    python -m pip install --upgrade pip
+    python -m pip install rich
+    python -m pip install numexpr
+    python -m pip install insightface
+    python -m pip install onnx
+    python -m pip install onnxruntime
+    python -m pip install onnxruntime-gpu
+    python -m pip install opencv-python
+    python -m pip install ifnude
+    python -m pip install cython
+    python -m pip install pytorch-lightning
+    python -m pip install blobfile
+
+    touch "${ROOT}/prepare-v1.ready"
+
+fi
+
+  
+if [ ! -f "${ROOT}/prepare-v2.ready" ]; then
+
+    python -m pip install wandb
+    python -m pip install fastai==1.0.60
+    python -m pip install tensorboardX
+    python -m pip install ffmpeg
+    python -m pip install ffmpeg-python
+    python -m pip install yt-dlp
+    python -m pip install opencv-python
+    python -m pip install Pillow
+
+    touch "${ROOT}/prepare-v2.ready"
+
+fi
+
+if [ ! -f "${ROOT}/prepare-v3.ready" ]; then
+
+    python -m pip install wandb
+    python -m pip install fastai==1.0.60
+    python -m pip install tensorboardX
+    python -m pip install ffmpeg
+    python -m pip install ffmpeg-python
+    python -m pip install yt-dlp
+    python -m pip install opencv-python
+    python -m pip install Pillow
+
+    touch "${ROOT}/prepare-v3.ready"
+
+fi
+
+if [ ! -f "${VOLUME_DIR_LOCAL}/styles.csv" ]; then
+
+    wget "https://huggingface.co/spaces/Moldwebs/srv/raw/main/sd-webui/styles.csv" -O ${VOLUME_DIR_LOCAL}/styles.csv
+
+fi
+
+if [ ! -f "${ROOT}/styles.csv" ]; then
+
+    cp ${VOLUME_DIR_LOCAL}/styles.csv ${ROOT}/styles.csv
+
+fi
+