improved UI

app.py

@@ -389,6 +389,7 @@ DEFAULT_SELECTION_METHOD = SELECTION_METHODS[0]
 #107, 119
 GENERATION_SEEDS = [42, 48, 56, 63, 74, 84, 86, 107, 119, 124, 125, 127, 128, 129]
 INTERNAL_SAMPLES = len(GENERATION_SEEDS)
+SHOW_SEED_CAPTIONS = False
 
 @spaces.GPU(duration=120)
 def generate(audio_path, num_display, guidance_scale, num_inference_steps, base_seed, selection_method=DEFAULT_SELECTION_METHOD):
@@ -444,7 +445,7 @@ def generate(audio_path, num_display, guidance_scale, num_inference_steps, base_
     else:
         best = select_best_images(pairs, int(num_display))
 
-    return [(img, f"Seed: {seed}") for img, seed in best], ""
+    return [(img, f"Seed: {seed}") if SHOW_SEED_CAPTIONS else img for img, seed in best], ""
 
 # ---------------------------------------------------------------------------
 # Model loading
@@ -530,7 +531,7 @@ def build_demo():
             "4. After a few seconds, the generated images will be displayed on the right."
         )
 
-        DEFAULT_NUM_DISPLAY = 14
+        DEFAULT_NUM_DISPLAY = 3
         DEFAULT_GUIDANCE_SCALE = 2.2
         DEFAULT_NUM_STEPS = 25
         DEFAULT_BASE_SEED = 42