added sdxl from sd community model

app.py

@@ -1,5 +1,5 @@
 import torch
-from diffusers import StableDiffusion3Pipeline, StableDiffusionPipeline, DiffusionPipeline
+from diffusers import StableDiffusion3Pipeline, StableDiffusionPipeline, DiffusionPipeline, DPMSolverSinglestepScheduler
 import gradio as gr
 import os
 import random
@@ -34,6 +34,12 @@ sdxl_pipe = DiffusionPipeline.from_pretrained(
 )
 sdxl_pipe.to(device)
 
+sdxl_flash_pipe = StableDiffusionXLPipeline.from_pretrained("sd-community/sdxl-flash", torch_dtype=torch.float16)
+sdxl_flash_pipe.to(device)
+
+# Ensure sampler uses "trailing" timesteps.
+sdxl_flash_pipe.scheduler = DPMSolverSinglestepScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing")
+
 # Define the image generation function for the Arena tab
 @spaces.GPU(duration=80)
 def generate_arena_images(
@@ -103,6 +109,8 @@ def generate_single_image(
         pipe = sd2_1_pipe
     elif model_choice == "sdxl":
         pipe = sdxl_pipe
+    elif model_choice == "sdxl flash":
+        pipe = sdxl_flash_pipe
     else:
         raise ValueError(f"Invalid model choice: {model_choice}")
 
@@ -192,13 +200,13 @@ with gr.Blocks(css=css) as demo:
                     )
                     model_choice_1 = gr.Dropdown(
                         label="Stable Diffusion Model 1",
-                        choices=["sd3 medium", "sd2.1", "sdxl"],
+                        choices=["sd3 medium", "sd2.1", "sdxl", "sdxl flash"],
                         value="sd3 medium",
                     )
                     model_choice_2 = gr.Dropdown(
                         label="Stable Diffusion Model 2",
-                        choices=["sd3 medium", "sd2.1", "sdxl"],
-                        value="sd2.1",
+                        choices=["sd3 medium", "sd2.1", "sdxl", "sdxl flash"],
+                        value="sdxl",
                     )
                     run_button = gr.Button("Run")
                 result_1 = gr.Gallery(label="Generated Images (Model 1)", elem_id="gallery_1")
@@ -301,7 +309,7 @@ with gr.Blocks(css=css) as demo:
                     )
                     model_choice = gr.Dropdown(
                         label="Stable Diffusion Model",
-                        choices=["sd3 medium", "sd2.1", "sdxl"],
+                        choices=["sd3 medium", "sd2.1", "sdxl", "sdxl flash"],
                         value="sd3 medium",
                     )
                     run_button = gr.Button("Run")