Update app.py

app.py

@@ -3,12 +3,15 @@ from diffusers import StableDiffusionPipeline
 import torch
 
 # Load the pre-trained model from Hugging Face (CompVis is the original model repository)
-pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", torch_dtype=torch.float16)
-pipe = pipe.to("cuda")
+pipe = StableDiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2", torch_dtype=torch.float16)
+
+# Ensure that the model runs on CPU
+pipe = pipe.to("cpu")  # Use CPU instead of CUDA (GPU)
 
 def generate_image(prompt):
     # Generate image based on the user's prompt
     image = pipe(prompt).images[0]
+
     return image
 
 # Create a Gradio interface