Update app.py

app.py

@@ -10,9 +10,9 @@ import numpy as np
 import PIL.Image
 import spaces
 import torch
-from diffusers import AutoencoderKL, DiffusionPipeline
+from diffusers import AutoencoderKL, StableDiffusionXLPipeline
 
-DESCRIPTION = "humblemikey/PixelWave10"
+DESCRIPTION = "# humblemikey/PixelWave10"
 if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>"
 
@@ -24,7 +24,7 @@ ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD") == "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(
+    pipe = StableDiffusionXLPipeline.from_pretrained(
         "humblemikey/PixelWave10",
         #vae=vae,
         torch_dtype=torch.float16,
@@ -66,8 +66,6 @@ def generate(
     return pipe(
         prompt=prompt,
         negative_prompt=negative_prompt,
-        prompt_2=prompt_2,
-        negative_prompt_2=negative_prompt_2,
         width=width,
         height=height,
         guidance_scale=guidance_scale_base,
@@ -145,7 +143,7 @@ with gr.Blocks(css="style.css") as demo:
                 minimum=10,
                 maximum=100,
                 step=1,
-                value=25,
+                value=40,
             )
 
     gr.Examples(