Update app.py

app.py

@@ -4,8 +4,8 @@ import torch
 # Load the pipeline
 pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2-1-base", torch_dtype=torch.float16)
 
-# If running on CUDA, ensure the model is moved to GPU for faster inference
-pipe.to("cuda")
+
+pipe.to("cpu")
 
 # Get the prompt from the user
 prompt = input("What do you want to see? ")