Create start.sh

start.sh

@@ -0,0 +1,42 @@
+#!/bin/bash
+
+echo "pod started"
+
+if [[ $PUBLIC_KEY ]]
+then
+    mkdir -p ~/.ssh
+    chmod 700 ~/.ssh
+    cd ~/.ssh
+    echo $PUBLIC_KEY >> authorized_keys
+    chmod 700 -R ~/.ssh
+    cd /
+    service ssh start
+fi
+
+#!/bin/bash
+if [[ -z "${HF_TOKEN}" ]] || [[ "${HF_TOKEN}" == "enter_your_huggingface_token_here" ]]
+then
+    echo "HF_TOKEN is not set"
+else
+    echo "HF_TOKEN is set, logging in..."
+    huggingface-cli login --token ${HF_TOKEN}
+fi
+
+
+# Start nginx as reverse proxy to enable api access
+service nginx start
+
+# Start JupyterLab
+jupyter lab --ip=0.0.0.0 --port=8888 --no-browser --allow-root --NotebookApp.allow_origin='*' &
+echo "JupyterLab started"
+
+# Check if there is a venv directory, if so, activate it
+if [ -d "/workspace/venv" ]; then
+    echo "venv directory found, activating it"
+    source /workspace/venv/bin/activate
+fi
+
+# Execute the user's script
+python3 /workspace/ComfyUI/main.py --listen
+
+sleep infinity