advanced settings

app.py

@@ -1,5 +1,7 @@
 from typing import Tuple
 
+import random
+import numpy as np
 import gradio as gr
 import spaces
 import torch
@@ -14,6 +16,8 @@ creating this amazing model, and a big thanks to [Gothos](https://github.com/Got
 for taking it to the next level by enabling inpainting with the FLUX.
 """
 
+MAX_SEED = np.iinfo(np.int32).max
+MAX_IMAGE_SIZE = 2048
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
 pipe = FluxInpaintPipeline.from_pretrained(
@@ -44,7 +48,15 @@ def resize_image_dimensions(
 
 
 @spaces.GPU()
-def process(input_image_editor, input_text, progress=gr.Progress(track_tqdm=True)):
+def process(
+    input_image_editor: dict,
+    input_text: str,
+    seed_slicer: int,
+    randomize_seed_checkbox: bool,
+    strength_slider: float,
+    num_inference_steps_slider: int,
+    progress=gr.Progress(track_tqdm=True)
+):
     if not input_text:
         gr.Info("Please enter a text prompt.")
         return None
@@ -64,14 +76,18 @@ def process(input_image_editor, input_text, progress=gr.Progress(track_tqdm=True
     resized_image = image.resize((width, height), Image.LANCZOS)
     resized_mask = mask.resize((width, height), Image.NEAREST)
 
+    if randomize_seed_checkbox:
+        seed_slicer = random.randint(0, MAX_SEED)
+    generator = torch.Generator().manual_seed(seed_slicer)
     return pipe(
         prompt=input_text,
         image=resized_image,
         mask_image=resized_mask,
         width=width,
         height=height,
-        strength=0.7,
-        num_inference_steps=24
+        strength=strength_slider,
+        generator=generator,
+        num_inference_steps=num_inference_steps_slider
     ).images[0], resized_mask
 
 
@@ -86,6 +102,7 @@ with gr.Blocks() as demo:
                 image_mode='RGB',
                 layers=False,
                 brush=gr.Brush(colors=["#FFFFFF"], color_mode="fixed"))
+
             with gr.Row():
                 input_text_component = gr.Text(
                     label="Prompt",
@@ -96,6 +113,35 @@ with gr.Blocks() as demo:
                 )
                 submit_button_component = gr.Button(
                     value='Submit', variant='primary', scale=0)
+
+            with gr.Accordion("Advanced Settings", open=False):
+                seed_slicer_component = gr.Slider(
+                    label="Seed",
+                    minimum=0,
+                    maximum=MAX_SEED,
+                    step=1,
+                    value=0,
+                )
+
+                randomize_seed_checkbox_component = gr.Checkbox(
+                    label="Randomize seed", value=True)
+
+                with gr.Row():
+                    strength_slider_component = gr.Slider(
+                        label="Strength",
+                        minimum=0,
+                        maximum=1,
+                        step=0.01,
+                        value=0.75,
+                    )
+
+                    num_inference_steps_slider_component = gr.Slider(
+                        label="Number of inference steps",
+                        minimum=1,
+                        maximum=50,
+                        step=1,
+                        value=20,
+                    )
         with gr.Column():
             output_gallery_component = gr.Gallery(
                 columns=[1],
@@ -110,7 +156,11 @@ with gr.Blocks() as demo:
         fn=process,
         inputs=[
             input_image_editor_component,
-            input_text_component
+            input_text_component,
+            seed_slicer_component,
+            randomize_seed_checkbox_component,
+            strength_slider_component,
+            num_inference_steps_slider_component
         ],
         outputs=[
             output_gallery_component