Upload folder using huggingface_hub

app.py

@@ -340,7 +340,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
@@ -348,25 +347,19 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 🌊 WaveSpeed AI 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(
-        """
+        """)
+    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",
@@ -376,7 +369,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",
@@ -400,13 +395,10 @@ 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="Recent Generations",
-            show_label=False,
-            elem_id="recent-gallery",
-            columns=4,
-            height="auto",
-        ).style(grid=2)
+        recent_gallery = gr.Gallery(label="Recent Generations",
+                                    columns=3,
+                                    rows=1,
+                                    show_label=True)
 
     def update_recent_gallery(prompt, results):
         recent_generations.append({
@@ -417,15 +409,12 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         })
         if len(recent_generations) > 8:
             recent_generations.pop(0)
-        
+
         # Flatten the images from each generation
         all_images = []
         for gen in reversed(recent_generations):  # Show newest first
-            all_images.extend([
-                gen["flux-dev"],
-                gen["hidream-dev"],
-                gen["hidream-full"]
-            ])
+            all_images.extend(
+                [gen["flux-dev"], gen["hidream-dev"], gen["hidream-full"]])
 
         # Update the Gradio gallery with the new list
         return recent_gallery.update(value=all_images)