Update app.py

app.py

@@ -1,31 +1,35 @@
-# app.py
-
 import streamlit as st
 from diffusers import StableDiffusionPipeline
 import torch
 from PIL import Image
 import io
+import os
+
+# Force CPU usage on Spaces
+os.environ["CUDA_VISIBLE_DEVICES"] = ""
 
 @st.cache_resource
 def load_model():
-    # Force CPU-only execution
     pipe = StableDiffusionPipeline.from_pretrained(
-        "runwayml/stable-diffusion-v1-5"
+        "runwayml/stable-diffusion-v1-5",
+        torch_dtype=torch.float32,
     )
-    pipe.to("cpu")
+    pipe.to("cpu")  # Force CPU usage
     return pipe
 
-st.title("🧠 AI Image Generator (CPU Only)")
+# Streamlit UI
+st.title("🎨 AI Image Generator (CPU - Hugging Face Spaces)")
 
 prompt = st.text_input("Enter your prompt:", 
-                       "A surreal multi-dimensional fractal landscape in vivid colors")
+                       "A multi-dimensional futuristic city with glowing lights, fractals, 8K")
 
-guidance = st.slider("Guidance scale (higher = more creative)", 1.0, 20.0, 7.5)
+guidance = st.slider("Creativity (Guidance Scale)", 1.0, 20.0, 7.5)
 
 if st.button("Generate"):
-    with st.spinner("Generating on CPU... please wait 2–5 minutes."):
+    with st.spinner("Generating image on CPU... please wait ⏳"):
         pipe = load_model()
         image = pipe(prompt, guidance_scale=guidance).images[0]
+
         st.image(image, caption="Generated Image", use_column_width=True)
 
         buf = io.BytesIO()