Add device check

app.py

@@ -6,6 +6,8 @@ from PIL import Image, ImageColor
 from diffusers import DDPMPipeline
 from diffusers import DDIMScheduler
 
+device = 'mps' if torch.backends.mps.is_available() else 'cuda' if torch.cuda.is_available() else 'cpu'
+
 # Load the pretrained pipeline
 pipeline_name = 'johnowhitaker/sd-class-wikiart-from-bedrooms'
 image_pipe = DDPMPipeline.from_pretrained(pipeline_name).to(device)