fix README

README.md

@@ -6,12 +6,13 @@ colorTo: red
 sdk: gradio
 sdk_version: 5.0.1
 app_file: app.py
-pinned: false
+pinned: true
 license: mit
 short_description: Arena for HiDream-I1-dev / -full and FLUX-dev
 models:
   - HiDream-ai/HiDream-I1-Full
   - HiDream-ai/HiDream-I1-Dev
+  - black-forest-labs/FLUX.1-dev
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -339,7 +339,6 @@ example_prompts = [
     "A sleek, futuristic sports car with glowing blue accents, racing through a virtual reality landscape, 3D render",
 ]
 
-
 # Use a state variable to store session ID
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     session_id = gr.State(None)  # Add this to store session ID
@@ -347,25 +346,19 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 🌊 WaveSpeedAI HiDream Arena")
 
     # Add the introduction with link to WaveSpeedAI
-    gr.Markdown(
-        """
+    gr.Markdown("""
         [WaveSpeedAI](https://wavespeed.ai/) is the global pioneer in accelerating AI-powered video and image generation.
-        "Our in-house inference accelerator provides lossless speedup on image & video generation based on our rich inference optimization software stack, including our in-house inference compiler, CUDA kernel libraries and parallel computing libraries."
-        """
-    )
-    gr.Markdown(
-        """
+        Our in-house inference accelerator provides lossless speedup on image & video generation based on our rich inference optimization software stack, including our in-house inference compiler, CUDA kernel libraries and parallel computing libraries.
+        """)
+    gr.Markdown("""
         This demo showcases the performance and outputs of leading image generation models, including HiDream and Flux, on our accelerated inference platform.
-        """
-    )
+        """)
 
     with gr.Row():
         with gr.Column(scale=3):
-            example_dropdown = gr.Dropdown(
-                choices=example_prompts,
-                label="Choose an example prompt",
-                interactive=True
-            )
+            example_dropdown = gr.Dropdown(choices=example_prompts,
+                                           label="Choose an example prompt",
+                                           interactive=True)
             input_text = gr.Textbox(
                 example_prompts[0],
                 label="Enter your prompt",
@@ -375,7 +368,9 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         with gr.Column(scale=1):
             generate_btn = gr.Button("Generate", variant="primary")
 
-    example_dropdown.change(lambda ex: ex, inputs=[example_dropdown], outputs=[input_text])
+    example_dropdown.change(lambda ex: ex,
+                            inputs=[example_dropdown],
+                            outputs=[input_text])
 
     # Two status boxes - small (default) and big (during generation)
     small_status_box = gr.Markdown("Ready to generate images",
@@ -398,8 +393,20 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
     performance_plot = gr.Plot(label="Performance Metrics")
 
-    with gr.Accordion("Recent Generations (last 8)", open=False):
-        recent_gallery = gr.Gallery(label="Prompt and Output", columns=3)
+    def get_recent_gallery_items():
+        gallery_items = []
+        for r in reversed(recent_generations):
+            gallery_items.append((r["flux-dev"], f"FLUX-dev: {r['prompt']}"))
+            gallery_items.append(
+                (r["hidream-dev"], f"HiDream-dev: {r['prompt']}"))
+            gallery_items.append(
+                (r["hidream-full"], f"HiDream-full: {r['prompt']}"))
+        return gallery_items
+
+    with gr.Accordion("Recent Generations (last 16)", open=False):
+        recent_gallery = gr.Gallery(get_recent_gallery_items(),
+                                    label="Prompt and Output",
+                                    columns=3)
 
     def update_recent_gallery(prompt, results):
         recent_generations.append({
@@ -408,16 +415,9 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             "hidream-dev": results["hidream-dev"],
             "hidream-full": results["hidream-full"],
         })
-        if len(recent_generations) > 8:
+        if len(recent_generations) > 16:
             recent_generations.pop(0)
-        gallery_items = []
-        for r in reversed(recent_generations):
-            gallery_items.append(
-                (r["flux-dev"], f"FLUX-dev: {r['prompt']}"))
-            gallery_items.append(
-                (r["hidream-dev"], f"HiDream-dev: {r['prompt']}"))
-            gallery_items.append(
-                (r["hidream-full"], f"HiDream-full: {r['prompt']}"))
+        gallery_items = get_recent_gallery_items()
         return gr.update(value=gallery_items)
 
     # Add custom CSS for the big status box