Update app.py

app.py

@@ -9,7 +9,7 @@ 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 = 'Skier8402/ddpm-celebahq-finetuned-butterflies-16epochs'
+pipeline_name = 'johnowhitaker/sd-class-wikiart-from-bedrooms'
 image_pipe = DDPMPipeline.from_pretrained(pipeline_name).to(device)
 
 # Set up the scheduler