small ux change: add random seed option

app.py

@@ -25,7 +25,6 @@ MAX_SEED = np.iinfo(np.int32).max
 device = "cuda"
 _dtype = torch.bfloat16
 hf_token = os.getenv("HF_TOKEN")
-
 print("init ovis_image")
 model_config = ovis_image_configs["ovis-image-7b"]
 ovis_image = OvisImageModel(model_config)
@@ -74,7 +73,10 @@ ovis_encoder = OvisEmbedder(
 
 
 @spaces.GPU(duration=75)
-def generate(prompt, img_height=1024, img_width=1024, seed=42, steps=50, guidance_scale=5.0):
+def generate(prompt, img_height=1024, img_width=1024, seed=42, randomize_seed=True, steps=50, guidance_scale=5.0):
+    if randomize_seed:
+        seed = random.randint(0, MAX_SEED)
+    
     print(f'inference with prompt : {prompt}, size: {img_height}x{img_width}, seed : {seed}, step : {steps}, cfg : {guidance_scale}')
     image = generate_image(
         device=next(ovis_image.parameters()).device,
@@ -169,12 +171,13 @@ with gr.Blocks(css=css) as demo:
                 )
 
             seed = gr.Slider(
-                label="Seed",
-                minimum=0,
-                maximum=MAX_SEED,
-                step=1,
-                value=42,
-            )
+                        label="Seed",
+                        minimum=0,
+                        maximum=MAX_SEED,
+                        step=1,
+                        value=0,
+                    )
+            randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
         
         gr.Examples(
             examples = examples,
@@ -187,7 +190,7 @@ with gr.Blocks(css=css) as demo:
     gr.on(
         triggers=[run_button.click, prompt.submit],
         fn = generate,
-        inputs = [prompt, img_height, img_width, seed, num_inference_steps, guidance_scale],
+        inputs = [prompt, img_height, img_width, seed, randomize_seed, num_inference_steps, guidance_scale],
         outputs = [result]
     )