Update app.py

device = 'cuda' if torch.cuda.is_available() else 'cpu'
pipe.to(device)

app.py

@@ -12,7 +12,9 @@ pipe = StableDiffusionXLPipeline.from_pretrained(
     variant="fp16",
     )
 pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
-pipe.to('cuda')
+device = 'cuda' if torch.cuda.is_available() else 'cpu'
+pipe.to(device)
+
 
 def launch(prompt, negative_prompt):
     prompt += " ,awesome, pixel art"