Update app.py

app.py

@@ -222,6 +222,7 @@ def get_duration(
     pixel_offset: int = 8,
     num_steps: int = 28,
     guidance: float = 3.5,
+    real_guidance: float = 4.5,
     seed: int = 42,
     randomize_seed: bool = False,
     progress=gr.Progress(track_tqdm=True)
@@ -243,6 +244,7 @@ def run_diptych_prompting(
     pixel_offset: int = 8,
     num_steps: int = 28,
     guidance: float = 3.5,
+    real_guidance: float = 4.5,
     seed: int = 42,
     randomize_seed: bool = False,
     progress=gr.Progress(track_tqdm=True)
@@ -296,7 +298,7 @@ def run_diptych_prompting(
         controlnet_conditioning_scale=ctrl_scale,
         guidance_scale=guidance,         
         negative_prompt="",
-        true_guidance_scale=guidance
+        true_guidance_scale=real_guidance
     ).images[0]
 
     # 5. Final cropping (logic from original script's main block)
@@ -327,7 +329,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 attn_enforce = gr.Slider(minimum=1.0, maximum=2.0, value=1.3, step=0.05, label="Attention Enforcement")
                 ctrl_scale = gr.Slider(minimum=0.5, maximum=1.0, value=0.95, step=0.01, label="ControlNet Scale")
                 num_steps = gr.Slider(minimum=20, maximum=50, value=28, step=1, label="Inference Steps")
-                guidance = gr.Slider(minimum=1.0, maximum=10.0, value=3.5, step=0.1, label="Guidance Scale")
+                guidance = gr.Slider(minimum=1.0, maximum=10.0, value=3.5, step=0.1, label="Distilled Guidance Scale")
+                real_guidance = gr.Slider(minimum=1.0, maximum=10.0, value=4.5, step=0.1, label="Real Guidance Scale")
                 width = gr.Slider(minimum=512, maximum=1024, value=768, step=64, label="Image Width")
                 height = gr.Slider(minimum=512, maximum=1024, value=768, step=64, label="Image Height")
                 pixel_offset = gr.Slider(minimum=0, maximum=32, value=8, step=1, label="Padding (Pixel Offset)")
@@ -349,7 +352,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
     run_button.click(
         fn=run_diptych_prompting,
-        inputs=[input_image, subject_name, target_prompt, attn_enforce, ctrl_scale, width, height, pixel_offset, num_steps, guidance, seed, randomize_seed],
+        inputs=[input_image, subject_name, target_prompt, attn_enforce, ctrl_scale, width, height, pixel_offset, num_steps, guidance, real_guidance, seed, randomize_seed],
         outputs=output_image
     )