Update app.py

app.py

@@ -226,12 +226,11 @@ MAX_IMAGE_SIZE = 4096
 
 if torch.cuda.is_available():
     vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
-    pipe = AutoPipelineForText2Image.from_pretrained(
-        "John6666/noobai-xl-nai-xl-epsilonpred10version-sdxl",
+    pipe = StableDiffusionXLPipeline.from_single_file(
+        "https://huggingface.co/bluepen5805/illustrious_pencil-XL/illustrious_pencil-XL-v2.0.0.safetensors",
         vae=vae,
-        torch_dtype=torch.float16,
         use_safetensors=True,
-        add_watermarker=False
+        torch_dtype=torch.float16,
     )
     pipe.to("cuda")