Update app.py

app.py

@@ -6,35 +6,33 @@ import torch
 from PIL import Image
 import io
 
-# Load model on CPU
+# Load model on CPU only
 @st.cache_resource
 def load_model():
     pipe = StableDiffusionPipeline.from_pretrained(
-        "runwayml/stable-diffusion-v1-5",  # You can change to another model if needed
-        torch_dtype=torch.float32          # Use float32 on CPU
+        "runwayml/stable-diffusion-v1-5",
+        torch_dtype=torch.float32,             # CPU needs float32
+        use_auth_token=True                    # If needed for private models
     )
-    return pipe.to("cpu")
+    pipe.to("cpu")                             # Ensure it's forced to CPU
+    return pipe
 
-# Streamlit UI
-st.title("🧠 AI Image Generator (CPU-Compatible)")
-st.markdown("Generate original, multidimensional images using Stable Diffusion — no GPU required!")
+# UI
+st.title("🎨 AI Image Generator (CPU Compatible)")
+st.markdown("No GPU? No problem. This app runs Stable Diffusion entirely on your CPU.")
 
-# Prompt and settings
-prompt = st.text_area("Enter your creative prompt:", 
-                      "A multi-dimensional alien city with glowing fractals, floating geometry, cosmic lighting, 8K resolution")
+prompt = st.text_area("Enter your prompt:", 
+                      "A surreal multi-dimensional alien forest with glowing trees and floating rocks, 8K")
 
-guidance = st.slider("Creativity (Guidance Scale)", 1.0, 20.0, 8.5)
+guidance = st.slider("Creativity (Guidance Scale)", 1.0, 20.0, 7.5)
 
-# Generate button
 if st.button("Generate Image"):
-    with st.spinner("Generating image. This may take a few minutes on CPU..."):
+    with st.spinner("Generating image on CPU... this may take 2-5 minutes."):
         pipe = load_model()
         image = pipe(prompt, guidance_scale=guidance).images[0]
 
         st.image(image, caption="Generated Image", use_column_width=True)
 
-        # Save for download
         buf = io.BytesIO()
         image.save(buf, format="PNG")
-        byte_im = buf.getvalue()
-        st.download_button("Download Image", byte_im, "generated.png", "image/png")
+        st.download_button("Download Image", buf.getvalue(), "generated.png", "image/png")