Update start.sh

start.sh

@@ -1,16 +1,13 @@
 #!/bin/bash
 
-# 1. Start the Python Guard Proxy in the background
-# This opens port 7860 immediately so Hugging Face sees the Space as "Live"
-python3 /home/user/app.py &
+# Optional: Print GPU info for debugging
+nvidia-smi
 
-# 2. Start the Stable Diffusion CPU Server in the background
-# We use a lightweight server (like fastsd-cpu or sd.cpp)
-# Here we assume you're using a python-based server for the backend
-python3 -m sd_backend_server --port 11434 --host 127.0.0.1 &
+# Optional: Pre-download model weights to the container cache 
+# to avoid timeout during the first user request
+# python -c "from diffusers import DiffusionPipeline; DiffusionPipeline.from_pretrained('runwayml/stable-diffusion-v1-5')"
 
-echo "Waiting for CPU Model to initialize..."
-sleep 10
+echo "Starting Stable Diffusion App..."
 
-# 3. Keep the container alive
-wait -n
+# Start the Python application
+python app.py