Update app.py

app.py

@@ -1,18 +1,17 @@
 import streamlit as st
-from diffusers import StableDiffusionXLPipeline
+from diffusers import AutoPipelineForText2Image
 import torch
 from io import BytesIO
 
-st.set_page_config(page_title="Dreamscape Visualizer (HD)")
-st.title("🌌 Dreamscape Visualizer – High Quality")
+st.set_page_config(page_title="Dreamscape Visualizer (Turbo)")
+st.title("🌌 Dreamscape Visualizer (Fast Mode)")
 
 @st.cache_resource
 def load_pipeline():
-    pipe = StableDiffusionXLPipeline.from_pretrained(
-        "stabilityai/stable-diffusion-xl-base-1.0",
+    pipe = AutoPipelineForText2Image.from_pretrained(
+        "stabilityai/sdxl-turbo",
         torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
-        variant="fp16" if torch.cuda.is_available() else None,
-        use_safetensors=True
+        variant="fp16" if torch.cuda.is_available() else None
     )
     pipe.to("cuda" if torch.cuda.is_available() else "cpu")
     return pipe
@@ -34,19 +33,20 @@ if st.button("Generate Dream Image"):
     if not prompt.strip():
         st.warning("Please describe your dream first!")
     else:
-        with st.spinner("Generating your high-res dream..."):
+        with st.spinner("Dreaming up your world..."):
             final_prompt = f"{prompt}, {style_modifiers[style]}"
-            result = pipe(prompt=final_prompt, guidance_scale=7.5, num_inference_steps=30)
+            result = pipe(prompt=final_prompt, guidance_scale=0.0, num_inference_steps=2)
             image = result.images[0]
-            st.image(image, caption="✨ Your Dream Visualized in HD", use_column_width=True)
+            st.image(image, caption="✨ Your Dream Visualized", use_column_width=True)
 
-            # Download Button
+            # Convert image to bytes for download
             buf = BytesIO()
             image.save(buf, format="PNG")
             byte_im = buf.getvalue()
+
             st.download_button(
                 label="📥 Download Dream Image",
                 data=byte_im,
-                file_name="dreamscape_hd.png",
+                file_name="dreamscape.png",
                 mime="image/png"
             )