Update app.py

app.py

@@ -12,7 +12,7 @@ fs = HfFileSystem(token=hf_token)
 api = HfApi()
 
 import torch
-from diffusers import DiffusionPipeline, AutoencoderKL
+from diffusers import StableDiffusionXLPipeline, AutoencoderKL
 
 device="cuda" if torch.cuda.is_available() else "cpu"
 
@@ -73,7 +73,7 @@ def infer (custom_model, weight_name, prompt, inf_steps, guidance_scale, width,
     
     vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
     
-    pipe = DiffusionPipeline.from_pretrained(
+    pipe = StableDiffusionXLPipeline.from_pretrained(
         "stabilityai/stable-diffusion-xl-base-1.0",
         vae=vae, torch_dtype=torch.float16, variant="fp16",
         use_safetensors=True