Add header on spaces

on_start.sh

@@ -26,6 +26,11 @@ function download-model() {
 
 ## ----------------------------
 
+## Adds a header to the webui on Hugging Face Spaces.
+sed -i -e '/demo:/r /app/stable-diffusion-webui/header_patch.py' /app/stable-diffusion-webui/modules/ui.py
+
+## ----------------------------
+
 ## Installing less models if $IS_SHARED_UI environment variable is set.
 if [ ${IS_SHARED_UI:-0} != 0 ]; then
     download-model --checkpoint "v1-5-pruned-emaonly.safetensors" "https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/39593d5650112b4cc580433f6b0435385882d819/v1-5-pruned-emaonly.safetensors"
@@ -41,7 +46,6 @@ if [ ${IS_SHARED_UI:-0} != 0 ]; then
     sed -i -e '/(train_interface, \"Train\", \"ti\"),/d' /app/stable-diffusion-webui/modules/ui.py
     sed -i -e '/extensions_interface, \"Extensions\", \"extensions\"/d' /app/stable-diffusion-webui/modules/ui.py
     sed -i -e '/settings_interface, \"Settings\", \"settings\"/d' /app/stable-diffusion-webui/modules/ui.py
-    sed -i -e '/demo:/r /app/stable-diffusion-webui/header_patch.py' /app/stable-diffusion-webui/modules/ui.py
     rm -rf /app/stable-diffusion-webui/scripts /app/stable-diffusion-webui/extensions/deforum-for-automatic1111-webui /app/stable-diffusion-webui/extensions/stable-diffusion-webui-images-browser /app/stable-diffusion-webui/extensions/sd-civitai-browser /app/stable-diffusion-webui/extensions/sd-webui-additional-networks
     cp -f shared-config.json config.json
     cp -f shared-ui-config.json ui-config.json