Update app.py

app.py

@@ -5,19 +5,19 @@ from PIL import Image
 
 # Title and description of the app
 st.title("🖼️ Stable Diffusion Image Generator")
-st.write("Generate images from text using the Stable Diffusion v1.5 model!")
+st.write("Generate images from text using the Stable Diffusion v2-1 model!")
 
 # Sidebar for user inputs
 st.sidebar.title("Input Options")
-prompt = st.sidebar.text_input("Enter your prompt", "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k")
+prompt = st.sidebar.text_input("Enter your prompt", "A futuristic cityscape at sunset, vibrant colors, 8k")
 generate_button = st.sidebar.button("Generate Image")
 
 # Load the pipeline when the app starts
 @st.cache_resource
 def load_pipeline():
-    # Use a smaller model or a more efficient pipeline
+    # Use a smaller and more efficient model
     pipe = StableDiffusionPipeline.from_pretrained(
-        "runwayml/stable-diffusion-v1-5", 
+        "stabilityai/stable-diffusion-2-1", 
         torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
         revision="fp16" if torch.cuda.is_available() else None
     )
@@ -49,6 +49,5 @@ if generate_button:
                     file_name="generated_image.png",
                     mime="image/png"
                 )
-                # print
         except Exception as e:
             st.error(f"An error occurred: {e}")