Update app.py

app.py

@@ -10,7 +10,7 @@ import numpy as np
 import PIL.Image
 import spaces
 import torch
-from diffusers import AutoencoderKL, DiffusionPipeline
+from diffusers import AutoencoderKL, DiffusionPipeline, StableDiffusionXLPipeline, KDPM2AncestralDiscreteScheduler
 
 MARKDOWN = """
 This demo utilizes <a href="https://huggingface.co/dataautogpt3/OpenDalleV1.1">OpenDalle V1.1</a> by @dataautogpt3.
@@ -34,8 +34,8 @@ ENABLE_REFINER = os.getenv("ENABLE_REFINER", "0") == "1"
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 if torch.cuda.is_available():
     vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
-    pipe = DiffusionPipeline.from_pretrained(
-        "dataautogpt3/OpenDalleV1.1",
+    pipe = StableDiffusionXLPipeline.from_pretrained(
+        "dataautogpt3/ProteusV0.2",
         vae=vae,
         torch_dtype=torch.float16,
         use_safetensors=True,