Update app.py

app.py

@@ -1,23 +1,24 @@
-# app.py
 import streamlit as st
 import torch
 import io
-from PIL import Image
 from diffusers import StableDiffusionPipeline
+from PIL import Image
+
+# Disable unnecessary warnings
+import warnings
+warnings.filterwarnings("ignore")
 
-# Configuration
+# Model configuration
 MODEL_ID = "CompVis/stable-diffusion-v1-4"
-DEFAULT_STEPS = 25
-DEFAULT_GUIDANCE = 7.5
 
 @st.cache_resource(show_spinner=False)
 def load_model():
     pipe = StableDiffusionPipeline.from_pretrained(
         MODEL_ID,
         torch_dtype=torch.float32,
-        use_safetensors=False,  # Force .bin files for compatibility
-        safety_checker=None,
-        requires_safety_checker=False
+        use_safetensors=False,  # Force using .bin files
+        use_auth_token=False,
+        safety_checker=None
     )
     pipe = pipe.to("cpu")
     pipe.enable_attention_slicing()
@@ -27,7 +28,7 @@ def main():
     st.set_page_config(page_title="Poetry to Image", page_icon="🎨")
     st.title("✨ Romantic Text to Image Converter")
     
-    # Session state initialization
+    # Initialize session state
     if 'generated_image' not in st.session_state:
         st.session_state.generated_image = None
     if 'error' not in st.session_state:
@@ -36,14 +37,14 @@ def main():
     # Sidebar controls
     with st.sidebar:
         st.header("Settings")
-        num_steps = st.slider("Inference Steps", 10, 50, DEFAULT_STEPS)
-        guidance_scale = st.slider("Guidance Scale", 3.0, 15.0, DEFAULT_GUIDANCE)
+        num_steps = st.slider("Inference Steps", 10, 50, 25)
+        guidance_scale = st.slider("Guidance Scale", 3.0, 15.0, 7.5)
 
     # Main interface
     prompt = st.text_area(
         "Enter your romantic text:",
         height=150,
-        placeholder="Example: 'Your eyes sparkle like stars...'"
+        placeholder="Example: 'Your smile shines brighter than the sun...'"
     )
 
     if st.button("Generate Image", type="primary"):
@@ -51,10 +52,10 @@ def main():
         st.session_state.error = None
         
         if not prompt.strip():
-            st.session_state.error = "Please enter some text!"
+            st.session_state.error = "Please enter some text first!"
         else:
             try:
-                with st.spinner("Creating artwork (3-8 minutes)..."):
+                with st.spinner("Creating artwork (5-10 minutes)..."):
                     pipe = load_model()
                     image = pipe(
                         prompt,