feat(gradio): add user-controlled seed with random mode for video generation

app_wip.py

@@ -3,6 +3,7 @@ import os
 import sys
 import uuid
 import shutil
+import random
 
 import gradio as gr
 import torch
@@ -166,6 +167,7 @@ def reward_forcing_inference(
     num_output_frames: int,
     use_ema: bool,
     output_root: str,
+    seed: int,
     progress: gr.Progress,
 ):
     """
@@ -180,7 +182,12 @@ def reward_forcing_inference(
 
     # --------------------- Device & randomness ---------------------
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-    set_seed(0)
+
+    if seed == -1:
+        seed = random.randint(0, 2**32 - 1)
+
+    set_seed(seed)
+    logs += f"Seed: {seed}\n"
 
     free_vram = get_cuda_free_memory_gb(device)
     logs += f"Free VRAM {free_vram} GB\n"
@@ -290,7 +297,11 @@ def reward_forcing_inference(
 
 @spaces.GPU(duration=200)
 def gradio_generate(
-    prompt: str, duration: str, use_ema: bool, progress=gr.Progress()
+    prompt: str,
+    duration: str,
+    use_ema: bool,
+    seed: int,
+    progress=gr.Progress(),
 ):
     """
     Triggered by Gradio:
@@ -316,6 +327,7 @@ def gradio_generate(
         num_output_frames=num_output_frames,
         use_ema=use_ema,
         output_root=OUTPUT_ROOT,
+        seed=int(seed),
         progress=progress,
     )
 
@@ -378,6 +390,8 @@ with gr.Blocks(title="Reward Forcing — Text-to-Video Demo") as demo:
 
         💡 This model performs best on **detailed prompts with multiple actions or transformations**.
 
+        🎲 Set a fixed seed for reproducible results, or use **-1** for a random seed each time.
+
         > ⏳ The first run may take a little longer while the model loads — generation is faster afterwards.
         """
     )
@@ -389,12 +403,6 @@ with gr.Blocks(title="Reward Forcing — Text-to-Video Demo") as demo:
             lines=4,
         )
 
-    gr.Examples(
-        examples=examples,
-        inputs=prompt_in,
-        label="Example prompts",
-    )
-
     with gr.Row():
         duration = gr.Radio(
             ["5s (21 frames)", "30s (120 frames)"],
@@ -402,6 +410,7 @@ with gr.Blocks(title="Reward Forcing — Text-to-Video Demo") as demo:
             label="Duration",
         )
         use_ema = gr.Checkbox(value=True, label="Use EMA weights (--use_ema)")
+        seed_in = gr.Number(value=-1, label="Seed (-1 = random)", precision=0)
 
     generate_btn = gr.Button("🚀 Generate Video", variant="primary")
 
@@ -409,9 +418,15 @@ with gr.Blocks(title="Reward Forcing — Text-to-Video Demo") as demo:
         video_out = gr.Video(label="Generated Video")
     logs_out = gr.Textbox(label="Logs", lines=12, interactive=False)
 
+    gr.Examples(
+        examples=examples,
+        inputs=prompt_in,
+        label="Example prompts",
+    )
+
     generate_btn.click(
         fn=gradio_generate,
-        inputs=[prompt_in, duration, use_ema],
+        inputs=[prompt_in, duration, use_ema, seed_in],
         outputs=[video_out, logs_out],
     )