Update app.py

app.py

@@ -81,6 +81,11 @@ DEFAULT_STEPS = {
     "Regular (30 steps)": 30,
 }
 
+# Default CFG for each mode
+DEFAULT_CFG = {
+    "Distilled (4 steps)": 1.0,
+    "Regular (30 steps)": 4.0,
+}
 
 def image_to_data_uri(img):
     buffered = io.BytesIO()
@@ -161,7 +166,7 @@ def update_dimensions_from_image(image_list):
 
 def update_steps_from_mode(mode_choice):
     """Update the number of inference steps based on the selected mode."""
-    return DEFAULT_STEPS[mode_choice]
+    return DEFAULT_STEPS[mode_choice], DEFAULT_CFG[mode_choice]
 
 
 @spaces.GPU(duration=85)
@@ -317,7 +322,7 @@ FLUX.2 [Klein] is a... [[model](https://huggingface.co/black-forest-labs/FLUX.2-
                             minimum=0.0,
                             maximum=10.0,
                             step=0.1,
-                            value=4.0,
+                            value=1.0,
                         )
                 
                 
@@ -354,7 +359,7 @@ FLUX.2 [Klein] is a... [[model](https://huggingface.co/black-forest-labs/FLUX.2-
     mode_choice.change(
         fn=update_steps_from_mode,
         inputs=[mode_choice],
-        outputs=[num_inference_steps]
+        outputs=[num_inference_steps, guidance_scale]
     )
 
     gr.on(