Update app.py

app.py

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