This PR adds the "Guidance Scale" parameter (#2)

- This PR adds the "Guidance Scale" parameter (6f60bb93b53a6a8d61dc33b8e77e171d98dd22d4)


Co-authored-by: Fabrice TIERCELIN <[email protected]>

app.py

@@ -19,7 +19,7 @@ pipe = FluxInpaintPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16).to(DEVICE)
 
 @spaces.GPU()
-def process(input_image_editor, uploaded_mask, input_text, strength, seed, randomize_seed, num_inference_steps, progress=gr.Progress(track_tqdm=True)):
+def process(input_image_editor, uploaded_mask, input_text, strength, seed, randomize_seed, num_inference_steps, guidance_scale=3.5, progress=gr.Progress(track_tqdm=True)):
     if not input_text:
         raise gr.Error("Please enter a text prompt.")
 
@@ -49,7 +49,8 @@ def process(input_image_editor, uploaded_mask, input_text, strength, seed, rando
         height=height,
         strength=strength,
         num_inference_steps=num_inference_steps,
-        generator=generator
+        generator=generator,
+        guidance_scale=guidance_scale
     ).images[0]
 
     return result, mask_image, seed
@@ -80,12 +81,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                     step=0.01,
                     label="Strength"
                 )
-                seed_number = gr.Number(
-                    label="Seed", 
-                    value=42,
-                    precision=0
-                )
-                randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
                 num_inference_steps = gr.Slider(
                     minimum=1,
                     maximum=100,
@@ -93,10 +88,23 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                     step=1,
                     label="Number of inference steps"
                 )
+                guidance_scale = gr.Slider(
+                    label="Guidance Scale",
+                    minimum=1,
+                    maximum=15,
+                    step=0.1,
+                    value=3.5,
+                )
+                seed_number = gr.Number(
+                    label="Seed", 
+                    value=42,
+                    precision=0
+                )
+                randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
             with gr.Accordion("Upload a mask", open=False):
                 uploaded_mask_component = gr.Image(label="Already made mask (black pixels will be preserved, white pixels will be redrawn)", sources=["upload"], type="pil")
             submit_button_component = gr.Button(
-                value='Generate', variant='primary')
+                value='Inpaint', variant='primary')
         with gr.Column(scale=1):
             output_image_component = gr.Image(
                 type='pil', image_mode='RGB', label='Generated image')
@@ -114,7 +122,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             strength_slider,
             seed_number,
             randomize_seed,
-            num_inference_steps
+            num_inference_steps,
+            guidance_scale
         ],
         outputs=[
             output_image_component,