Update app.py

app.py

@@ -4,12 +4,12 @@ import gradio as gr
 from diffusers import StableDiffusionImg2ImgPipeline
 
 # Load the Stable Diffusion img2img pipeline
-device = "cuda"
+device = "cpu"
 pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
     "runwayml/stable-diffusion-v1-5",
     torch_dtype=torch.float16,
 ).to(device)
-pipe.enable_attention_slicing()
+#pipe.enable_attention_slicing()
 
 # Predefined prompts
 PROMPTS = {