Update app.py

app.py

@@ -2,8 +2,8 @@ import torch
 import gradio as gr
 from diffusers import StableDiffusionXLPipeline, DPMSolverSinglestepScheduler
 
-# Load model and set the device to GPU
-pipe = StableDiffusionXLPipeline.from_pretrained("sd-community/sdxl-flash", torch_dtype=torch.float16).to("cuda")
+# Load model and set the device to CPU
+pipe = StableDiffusionXLPipeline.from_pretrained("sd-community/sdxl-flash", torch_dtype=torch.float32).to("cpu")
 
 # Ensure sampler uses "trailing" timesteps
 pipe.scheduler = DPMSolverSinglestepScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing")