add randomized seed option (#3)

- add randomized seed option (1e32b52544ac501fbf171a55abb15fa5fc976d1b)


Co-authored-by: Linoy Tsaban <[email protected]>

app.py

@@ -17,12 +17,17 @@ pipeline.load_lora_weights(
     hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8steps-lora.safetensors"), adapter_name="hyper-sd"
 )
 pipeline.set_adapters(["hyper-sd"], adapter_weights=[0.125])
-
+MAX_SEED = np.iinfo(np.int32).max
+def get_seed(randomize_seed: bool, seed: int) -> int:
+    """
+    Get the random seed.
+    """
+    return np.random.randint(0, MAX_SEED) if randomize_seed else seed
 #####################################
 #  The function for our Gradio app  #
 #####################################
 @spaces.GPU(duration=120)
-def generate(prompt, input_image, progress=gr.Progress(track_tqdm=True)):
+def generate(prompt, input_image, seed, progress=gr.Progress(track_tqdm=True)):
     """
     Runs the Flux Control pipeline for editing the given `input_image`
     with the specified `prompt`. The pipeline is on CPU by default.
@@ -35,7 +40,7 @@ def generate(prompt, input_image, progress=gr.Progress(track_tqdm=True)):
         max_sequence_length=512,
         height=input_image.height,
         width=input_image.width,
-        generator=torch.manual_seed(0)
+        generator=torch.manual_seed(seed)
     ).images[0]
 
     return output_image
@@ -65,14 +70,19 @@ def launch_app():
                         label="Your edit prompt",
                         placeholder="e.g. 'Turn the color of the mushroom to blue'"
                     )
+                    randomize_seed = gr.Checkbox(label="Randomize Seed", value=True)
                     generate_button = gr.Button("Generate")
         
             output_image = gr.Image(label="Edited Image")
 
         # Connect button to function
         generate_button.click(
+        get_seed,
+        inputs=[randomize_seed, seed],
+        outputs=[seed],
+    ).then(
             fn=generate,
-            inputs=[prompt, input_image],
+            inputs=[prompt, input_image, seed],
             outputs=[output_image],
         )