updated

app.py

@@ -40,7 +40,8 @@ from diffusers import StableDiffusionImg2ImgPipeline
 # Load Stable Diffusion 2.1 img2img model
 model_id = "stabilityai/stable-diffusion-2-1"
 pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id, torch_dtype=torch.float32)  
-pipe.to("cpu")  # Change to "cuda" for GPU
+# pipe.to("cpu")  # Change to "cuda" for GPU
+pipe.to("cuda" if torch.cuda.is_available() else "cpu")
 
 # Enable CPU optimizations
 pipe.enable_attention_slicing()