Update app.py

app.py

@@ -32,10 +32,11 @@ def run(prompt, negative_prompt="", guidance_scale=7.0, pag_scale=3.0, pag_layer
         guidance_scale = 0.0
         
     generator = torch.Generator(device="cuda").manual_seed(seed)
-    image_pag = pipe(prompt, guidance_scale=guidance_scale, pag_scale=pag_scale, pag_applied_layers=pag_layers, generator=generator, num_inference_steps=25).images[0]    
+    image_pag = pipe(prompt, negative_prompt=negative_prompt, guidance_scale=guidance_scale, pag_scale=pag_scale, pag_applied_layers=pag_layers, generator=generator, num_inference_steps=25).images[0]    
     
     generator = torch.Generator(device="cuda").manual_seed(seed)
-    image_normal = pipe(prompt, guidance_scale=guidance_scale, generator=generator, num_inference_steps=25).images[0]
+    image_normal = pipe(prompt, negative_prompt=negative_prompt, guidance_scale=guidance_scale, generator=generator, num_inference_steps=25).images[0]
+    
     return (image_pag, image_normal), seed
 
 css = '''
@@ -55,7 +56,7 @@ with gr.Blocks(css=css, theme=theme) as demo:
         button = gr.Button("Generate", min_width=120)
       output = ImageSlider(label="Left: PAG, Right: No PAG", interactive=False)
       with gr.Accordion("Advanced Settings", open=False):
-        guidance_scale = gr.Number(label="Guidance Scale", value=7.0)
+        guidance_scale = gr.Number(label="CFG Guidance Scale", info="The guidance scale for CFG, ignored if no prompt is entered (unconditional generation)", value=7.0)
         negative_prompt = gr.Textbox(label="Negative prompt", info="Is only applied for the CFG part, leave blank for unconditional generation")
         pag_scale = gr.Number(label="Pag Scale", value=3.0)
         pag_layers = gr.Dropdown(label="Model layers to apply Pag to", info="mid is the one used on the paper, up and down blocks seem unstable", choices=["up", "mid", "down"], multiselect=True, value="mid")