Update app.py

app.py

@@ -1,12 +1,11 @@
 import streamlit as st
 import torch
 from diffusers import AutoPipelineForImage2Image
-from diffusers.utils import load_image
 from PIL import Image
 
 st.title("Stable Diffusion Image-to-Image")
 
-# Load the pipeline (no caching)
+# Load the pipeline
 pipeline = AutoPipelineForImage2Image.from_pretrained(
     "stabilityai/stable-diffusion-2-1",
     torch_dtype=torch.float16,
@@ -14,7 +13,7 @@ pipeline = AutoPipelineForImage2Image.from_pretrained(
     use_safetensors=True
 )
 pipeline.enable_model_cpu_offload()
-
+# pipeline.enable_xformers_memory_efficient_attention()  # ❌ REMOVE OR COMMENT THIS LINE
 
 # Upload an image
 uploaded_file = st.file_uploader("Upload an image", type=["png", "jpg", "jpeg"])
@@ -28,4 +27,4 @@ if uploaded_file:
     if st.button("Generate Image"):
         with st.spinner("Generating..."):
             image = pipeline(prompt, image=init_image).images[0]
-            st.image(image, caption="Generated Image", use_column_width=True) 
+            st.image(image, caption="Generated Image", use_column_width=True)