Update app.py

app.py

@@ -5,17 +5,17 @@ from PIL import Image
 
 # Streamlit App Title
 st.title("🖼️ Stable Diffusion Image Generator")
-st.write("Generate images from text prompts using Stable Diffusion!")
+st.write("Generate images from text prompts using Stable Diffusion 2.1!")
 
 # 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 model to load it only once
+# Cache the Stable Diffusion 2.1 model to load it only once
 @st.cache_resource
 def load_model():
-    model = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
+    model = StableDiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2-1", torch_dtype=torch.float16)
     model = model.to("cuda" if torch.cuda.is_available() else "cpu")
     return model