Update app.py

app.py

@@ -63,7 +63,8 @@ def process_image(
                 save_masked_image=False,
                 output_path="",
                 learning_rate=learning_rate,
-                max_steps=max_steps
+                max_steps=max_steps,
+                optimization_steps=optimization_steps,
             ).images[0]
             pipe.vae = pipe.vae.to(torch.float16)
         return result, f"✅ Inpainting completed with seed {seed}."
@@ -111,7 +112,7 @@ with gr.Blocks() as demo:
         """
     )
     gr.Markdown("<h1>🍲 FlowChef 🍲</h1>")
-    gr.Markdown("<h2>Inversion/Gradient/Training-free Steering of Flux.1[Dev]</h2>")
+    gr.Markdown("<h2>Inversion/Gradient/Training-free Steering of <u>Flux.1[dev]</u></h2>")
     gr.Markdown("<h2><p><a href='https://flowchef.github.io/'>Project Page</a> | <a href='#'>Paper</a></p> (Steering Rectified Flow Models in the Vector Field for Controlled Image Generation)</h2>")
     gr.Markdown("<h3>💡 We recommend going through our <a href='#'>tutorial introduction</a> before getting started!</h3>")
 
@@ -288,7 +289,7 @@ with gr.Blocks() as demo:
                 30,  # max_steps
                 1.0,  # learning_rate
                 20,  # max_source_steps
-                10,  # optimization_steps
+                5,  # optimization_steps
                 2,  # true_cfg
             ],
             [
@@ -304,7 +305,7 @@ with gr.Blocks() as demo:
                 20,  # max_steps
                 1.0,  # learning_rate
                 20,  # max_source_steps
-                10,  # optimization_steps
+                5,  # optimization_steps
                 2,  # true_cfg
             ],
             [
@@ -352,7 +353,7 @@ with gr.Blocks() as demo:
                 30,  # max_steps
                 1,  # learning_rate
                 20,  # max_source_steps
-                10,  # optimization_steps
+                3,  # optimization_steps
                 2,  # true_cfg
             ],
             [
@@ -368,7 +369,7 @@ with gr.Blocks() as demo:
                 20,  # max_steps
                 1,  # learning_rate
                 20,  # max_source_steps
-                5,  # optimization_steps
+                3,  # optimization_steps
                 2,  # true_cfg
             ],
             [