Update app.py

app.py

@@ -1,5 +1,5 @@
 import torch
-from diffusers import StableDiffusion3Pipeline, StableDiffusion2Pipeline, StableDiffusionXLBasePipeline
+from diffusers import StableDiffusion3Pipeline, StableDiffusionPipeline, DiffusionPipeline
 import gradio as gr
 import os
 import random
@@ -26,12 +26,12 @@ sd3_medium_pipe = StableDiffusion3Pipeline.from_pretrained(
 )
 sd3_medium_pipe.to(device)
 
-sd2_1_pipe = StableDiffusion2Pipeline.from_pretrained(
+sd2_1_pipe = StableDiffusionPipeline.from_pretrained(
     "stabilityai/stable-diffusion-2-1", torch_dtype=torch.float16
 )
 sd2_1_pipe.to(device)
 
-sdxl_pipe = StableDiffusionXLBasePipeline.from_pretrained(
+sdxl_pipe = DiffusionPipeline.from_pretrained(
     "stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16
 )
 sdxl_pipe.to(device)