Update app.py

app.py

@@ -1,10 +1,10 @@
 import spaces
 import torch
-from diffusers import PixArtSigmaPipeline
+from custom_pipeline import FluxPipeline
 import gradio as gr
 
 # Load the pre-trained diffusion model
-pipe = PixArtSigmaPipeline.from_pretrained('ptx0/pixart-900m-1024-ft', torch_dtype=torch.bfloat16)
+pipe = FluxPipeline.from_pretrained('terminusresearch/flux-booru-cfg3.5', torch_dtype=torch.bfloat16)
 pipe.to('cuda')
 import re
 
@@ -31,9 +31,9 @@ def generate(prompt, guidance_scale, num_inference_steps, resolution, negative_p
 
 # Example prompts to demonstrate the model's capabilities
 example_prompts = [
- ["A futuristic cityscape at night under a starry sky", 7.5, 25, "blurry, overexposed"],
- ["A serene landscape with a flowing river and autumn trees", 8.0, 20, "crowded, noisy"],
- ["An abstract painting of joy and energy in bright colors", 9.0, 30, "dark, dull"]
+ ["A futuristic cityscape at night under a starry sky", 3.5, 25, "blurry, overexposed"],
+ ["A serene landscape with a flowing river and autumn trees", 3, 20, "crowded, noisy"],
+ ["An abstract painting of joy and energy in bright colors", 3.0, 30, "dark, dull"]
 ]
 
 # Create a Gradio interface, 1024x1024,1152x960,896x1152
@@ -50,7 +50,7 @@ iface = gr.Interface(
  examples=example_prompts,
  title="PixArt 900M",
  description=(
- "This is a 900M parameter model expanded from PixArt Sigma 1024px (600M) by adding 14 layers to deepen the transformer."
+ "This is a Flux-based 12B parameter model, fully trained across 8xH100 GPUs to reintroduce classifier-free guidance (CFG) sampling."
  "<br />This model is being <strong>actively trained</strong> on 3.5M samples across a wide distribution of photos, synthetic data, cinema, anime, and safe-for-work furry art."
  "<br />"
  "<br />&nbsp;The datasets been filtered for extremist and illegal content, but it is possible to produce toxic outputs. <strong>This model has not been safety-aligned or fine-tuned</strong>."
@@ -64,8 +64,7 @@ iface = gr.Interface(
  "<br />"
  "<ul>"
  "<li>Lead trainer: @pseudoterminalx (bghira@GitHub)</li>"
- "<li>Architecture: @jimmycarter (AmericanPresidentJimmyCarter@GitHub)</li>"
- "<li>Datasets: @ProGamerGov, @jimmycarter, @pseudoterminalx</li>"
+ "<li>Datasets: @jimmycarter, @pseudoterminalx</li>"
  "</ul>"
  )
 ).launch()