Update app.py

app.py

@@ -18,6 +18,8 @@ pipe = StableDiffusionXLPipeline.from_pretrained(
 pipe.unet = register_cross_attention_hook(pipe.unet)
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 pipe = pipe.to(device)
+if device == 'cpu':
+    pipe = pipe.to(dtype=torch.float32)
 
 
 def inference(prompt):