Update app.py

app.py

@@ -1,10 +1,10 @@
 import torch
-from diffusers import StableDiffusion3Pipeline
+from diffusers import PixArtSigmaPipeline
 import gradio as gr
 import spaces
 
 # Load the pre-trained diffusion model
-pipe = StableDiffusion3Pipeline.from_pretrained('ptx0/sd3-reality-mix', torch_dtype=torch.bfloat16)
+pipe = PixArtSigmaPipeline.from_pretrained('ptx0/pixart-900m-1024-ft', torch_dtype=torch.bfloat16)
 pipe.to('cuda')
 import re
 
@@ -48,6 +48,6 @@ iface = gr.Interface(
     ],
     outputs=gr.Gallery(height=1024, min_width=1024, columns=2),
     examples=example_prompts,
-    title="SD3 Reality Mix",
-    description="SD3 Reality Mix is an attempt to finetune SD3 Medium 2B on photo-concept-bucket, a 576k photo set of images."
+    title="PixArt 900M",
+    description="This is a 900M parameter model expanded from PixArt Sigma 1024px (600M) by adding 14 layers to deepen the transformer. It is being actively trained on 3.5M samples across a wide distribution of photos, synthetic data, cinema, anime, and safe-for-work furry art. The datasets been filtered for extremist and illegal content, but it is possible to produce toxic outputs. Be mindful when using this demo space that you do not inadvertently share images without adequate preparation and informing the users that these images are AI generated.<br /><br />This model is being trained with support from Fal.ai's grant program. See https://fal.ai/grants for more information."
 ).launch()