Update app.py

app.py

@@ -9,7 +9,7 @@ pipe.to('cuda')
 
 # Define the image generation function with adjustable parameters and a progress bar
 @spaces.GPU
-def generate(prompt, guidance_scale, num_inference_steps, negative_prompt):
+def generate(prompt, guidance_scale, guidance_rescale, num_inference_steps, negative_prompt):
     with gr.Progress(steps=num_inference_steps) as progress:
         for i in range(num_inference_steps):
             progress.update(progress=i)
@@ -17,6 +17,7 @@ def generate(prompt, guidance_scale, num_inference_steps, negative_prompt):
             prompt,
             negative_prompt=negative_prompt,
             guidance_scale=guidance_scale,
+            guidance_rescale=guidance_rescale,
             num_inference_steps=num_inference_steps
         ).images
 
@@ -32,7 +33,8 @@ iface = gr.Interface(
     fn=generate,
     inputs=[
         gr.Text(label="Enter your prompt"),
-        gr.Slider(5, 10, step=0.1, label="Guidance Scale", value=7.5),
+        gr.Slider(5, 10, step=0.1, label="Guidance Scale", value=11.5),
+        gr.Slider(5, 10, step=0.1, label="Rescale classifier-free guidance", value=0.7),
         gr.Slider(10, 50, step=5, label="Number of Inference Steps", value=25),
         gr.Text(value="underexposed, blurry, ugly, washed-out", label="Negative Prompt")
     ],