Randomize the seed

app.py

@@ -5,6 +5,7 @@ import tempfile
 import gradio as gr
 import numpy as np
 import torch
+import random
 from glob import glob
 from torchvision.transforms import CenterCrop, Compose, Resize
 
@@ -175,6 +176,7 @@ def fn_vis_camera(camera_args):
             gr.update(visible=vis_prompt), \
             gr.update(visible=vis_num_samples), \
             gr.update(visible=vis_seed), \
+            gr.update(visible=vis_seed), \
             gr.update(visible=vis_start), \
             gr.update(visible=vis_gen_video, value=None), \
             gr.update(visible=vis_repeat_highlight)
@@ -388,6 +390,7 @@ def visualized_camera_poses(step2_camera_motion):
             gr.update(visible=vis_prompt), \
             gr.update(visible=vis_num_samples), \
             gr.update(visible=vis_seed), \
+            gr.update(visible=vis_seed), \
             gr.update(visible=vis_start), \
             gr.update(visible=vis_gen_video), \
             gr.update(visible=vis_repeat_highlight)
@@ -503,12 +506,15 @@ def process_input_image(input_image, resize_mode):
             gr.update(visible=vis_prompt), \
             gr.update(visible=vis_num_samples), \
             gr.update(visible=vis_seed), \
+            gr.update(visible=vis_seed), \
             gr.update(visible=vis_start), \
             gr.update(visible=vis_gen_video), \
             gr.update(visible=vis_repeat_highlight)
 
-def model_run(input_image, fps_id, seed, n_samples, camera_args):
+def model_run(input_image, fps_id, randomize_seed, seed, n_samples, camera_args):
     global model, device, camera_dict, num_frames, num_steps, width, height
+    if randomize_seed:
+        seed = random.randint(0, 2**63 - 1)
     RT = process_camera(camera_dict, camera_args, num_frames=num_frames, width=width, height=height).reshape(-1,12)
 
     video_path = motionctrl_sample(
@@ -657,14 +663,15 @@ def main(args):
                     generation_dec = gr.Markdown(f"\n 1. Set `FPS`.; \
                                                     \n 2. Set `n_samples`; \
                                                     \n 3. Set `seed`; \
-                                                    \n 4. Click `Start generation !` to generate videos; ", visible=False)
+                                                    \n 4. Click `Start generation!` to generate videos; ", visible=False)
                     # prompt = gr.Textbox(value="a dog sitting on grass", label="Prompt", interactive=True, visible=False)
                     prompt = gr.Slider(minimum=5, maximum=30, step=1, label="FPS", value=10, visible=False)
                     n_samples = gr.Number(value=1, precision=0, interactive=True, label="n_samples", visible=False)
-                    seed = gr.Number(value=1234, precision=0, interactive=True, label="Seed", visible=False)
-                    start = gr.Button(value="Start generation !", visible=False)
+                    randomize_seed = gr.Checkbox(label = "\U0001F3B2 Randomize seed", value = True, info = "If checked, result is always different", interactive=True, visible=False)
+                    seed = gr.Slider(label="Seed", minimum=0, maximum=2**63 - 1, step=1, randomize=True, interactive=True, visible=False)
+                    start = gr.Button(value="Start generation!", variant = "primary", visible=False)
                 with gr.Column():
-                    gen_video = gr.Video(value=None, label="Generate Video", visible=False)
+                    gen_video = gr.Video(value=None, label="Generated Video", visible=False)
                     repeat_highlight=gr.HighlightedText(value=[("",""), (f"1. If the motion control is not obvious, try to increase the `Motion Speed`. \
                                                 \n 2. If the generated videos are distored severely, try to descrease the `Motion Speed` \
                                                 or increase `FPS`.", "Normal")],
@@ -699,7 +706,7 @@ def main(args):
                 generation_dec, 
                 prompt, 
                 n_samples, 
-                seed, start, gen_video, repeat_highlight])
+                randomize_seed, seed, start, gen_video, repeat_highlight])
 
         keep_spatial_raition_botton.click(
             fn=process_input_image, 
@@ -730,7 +737,7 @@ def main(args):
                 generation_dec, 
                 prompt, 
                 n_samples, 
-                seed, start, gen_video, repeat_highlight])
+                randomize_seed, seed, start, gen_video, repeat_highlight])
         
 
         camera_info.click(
@@ -751,7 +758,7 @@ def main(args):
                      camera_args, 
                      camera_reset, camera_vis,
                      vis_camera,
-                     step3_prompt_generate, generation_dec, prompt, n_samples, seed, start, gen_video, repeat_highlight],
+                     step3_prompt_generate, generation_dec, prompt, n_samples, randomize_seed, seed, start, gen_video, repeat_highlight],
         )
 
 
@@ -793,6 +800,7 @@ def main(args):
                                   generation_dec,
                                   prompt, 
                                   n_samples, 
+                                  randomize_seed,
                                   seed, 
                                   start, 
                                   gen_video,
@@ -809,7 +817,7 @@ def main(args):
 
 
         start.click(fn=model_run, 
-                    inputs=[process_image, prompt, seed, n_samples, camera_args], 
+                    inputs=[process_image, prompt, randomize_seed, seed, n_samples, camera_args], 
                     outputs=gen_video)
 
         # set example