Update app.py

app.py

@@ -27,6 +27,7 @@ def infer(
     seed=42,
     randomize_seed=False,
     guidance_scale=4.0,
+    true_guidance_scale=1.0,
     num_inference_steps=50,
     progress=gr.Progress(track_tqdm=True),
 ):
@@ -53,7 +54,7 @@ def infer(
         negative_prompt=negative_prompt,
         num_inference_steps=num_inference_steps,
         generator=generator,
-        true_cfg_scale=1.0, # Use a fixed default for distilled guidance
+        true_cfg_scale=true_guidance_scale,
         guidance_scale=guidance_scale  
     ).images[0]
 
@@ -103,13 +104,21 @@ with gr.Blocks(css=css) as demo:
 
             with gr.Row():
                 guidance_scale = gr.Slider(
-                    label="Guidance scale",
+                    label="Distilled guidance scale",
                     minimum=0.0,
                     maximum=10.0,
                     step=0.1,
                     value=4.0,
                 )
 
+                true_guidance_scale = gr.Slider(
+                    label="True guidance scale",
+                    minimum=1.0,
+                    maximum=10.0,
+                    step=0.1,
+                    value=1.0
+                )
+
                 num_inference_steps = gr.Slider(
                     label="Number of inference steps",
                     minimum=1,
@@ -130,6 +139,7 @@ with gr.Blocks(css=css) as demo:
             seed,
             randomize_seed,
             guidance_scale,
+            true_guidance_scale,
             num_inference_steps,
         ],
         outputs=[result, seed],