Update app.py

app.py

@@ -1,21 +1,21 @@
 import streamlit as st
-from diffusers import StableDiffusionPipeline
+from diffusers import DiffusionPipeline
 import torch
 from PIL import Image
 
 # Streamlit App Title
 st.title("🖼️ Stable Diffusion Image Generator")
-st.write("Generate images from text prompts using Stable Diffusion 2.1!")
+st.write("Generate images from text prompts using Stable Diffusion!")
 
 # Sidebar for prompt input and button
 st.sidebar.title("Image Generation Settings")
 prompt = st.sidebar.text_input("Enter your prompt", "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k")
 generate_button = st.sidebar.button("Generate Image")
 
-# Cache the Stable Diffusion 2.1 model to load it only once
+# Cache the Stable Diffusion model to load it only once
 @st.cache_resource
 def load_model():
-    model = StableDiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2-1", torch_dtype=torch.float16)
+    model = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
     model = model.to("cuda" if torch.cuda.is_available() else "cpu")
     return model