Add an option to sample random seed

app.py

@@ -53,6 +53,8 @@ def create_advanced_demo(model: Model) -> gr.Blocks:
                                           step=1,
                                           value=20,
                                           label='Number of Steps')
+                    randomize_seed = gr.Checkbox(value=False,
+                                                 label='Randomize Seed')
                     seed = gr.Slider(0,
                                      100000,
                                      step=1,
@@ -83,9 +85,13 @@ def create_advanced_demo(model: Model) -> gr.Blocks:
                              model_name,
                              scheduler_type,
                              num_steps,
+                             randomize_seed,
                              seed,
                          ],
-                         outputs=result)
+                         outputs=[
+                             result,
+                             seed,
+                         ])
     return demo
 
 

model.py

@@ -106,12 +106,15 @@ class Model:
         model_name: str,
         scheduler_type: str,
         num_steps: int,
+        randomize_seed: bool,
         seed: int,
-    ) -> PIL.Image.Image:
+    ) -> tuple[PIL.Image.Image, int]:
         self.set_pipeline(model_name, scheduler_type)
         if scheduler_type == 'PNDM':
             num_steps = max(4, min(num_steps, 100))
-        return self.generate(seed, num_steps)[0]
+        if randomize_seed:
+            seed = self.rng.randint(0, 100000)
+        return self.generate(seed, num_steps)[0], seed
 
     @staticmethod
     def to_grid(images: list[PIL.Image.Image],