Add guidance_scale as parameter to gradio app

app.py

@@ -12,13 +12,14 @@ dreambooth_model = keras_cv.models.StableDiffusion(
 loaded_diffusion_model = from_pretrained_keras("melanit/dreambooth_voyager_v2")
 dreambooth_model._diffusion_model = loaded_diffusion_model
 
-def generate_images(prompt: str, negative_prompt:str, batch_size: int, num_steps: int):
+def generate_images(prompt: str, negative_prompt:str, batch_size: int, num_steps: int, guidance_scale: float):
     """
     This function will infer the trained dreambooth (stable diffusion) model
     Args:
         prompt (str): The input text
         batch_size (int): The number of images to be generated
         num_steps (int): The number of denoising steps
+        guidance_scale (float): The Guidance Scale
     Returns:
         outputs (List): List of images that were generated using the model
     """
@@ -27,6 +28,7 @@ def generate_images(prompt: str, negative_prompt:str, batch_size: int, num_steps
         negative_prompt=negative_prompt,
         batch_size=batch_size,
         num_steps=num_steps,
+        unconditional_guidance_scale=guidance_scale
     )
    
     return outputs
@@ -39,14 +41,15 @@ with gr.Blocks() as demo:
             negative_prompt = gr.Textbox(lines=1, value="", label="Negative Prompt")
             samples = gr.Slider(minimum=1, maximum=10, value=1, step=1, label="Number of Images")
             num_steps = gr.Slider(minimum=1, maximum=100, value=50, step=1, label="Denoising Steps")
+            guidance_scale = gr.Slider(value=7.5, step=0.5, label="Guidance scale")
             run = gr.Button(value="Run")
         with gr.Column():
             gallery = gr.Gallery(label="Outputs").style(grid=(1,2))
 
-    run.click(generate_images, inputs=[prompt,negative_prompt, samples, num_steps], outputs=gallery)
+    run.click(generate_images, inputs=[prompt, negative_prompt, samples, num_steps, guidance_scale], outputs=gallery)
     
-    gr.Examples([["photo of voyager spaceship in space, high quality, blender, 3d, trending on artstation, 8k","bad, ugly, malformed, deformed, out of frame, blurry", 1, 50]],
-                [prompt,negative_prompt, samples,num_steps], gallery, generate_images)
+    gr.Examples([["photo of voyager spaceship in space, high quality, blender, 3d, trending on artstation, 8k","bad, ugly, malformed, deformed, out of frame, blurry", 1, 50, 7.5]],
+                [prompt, negative_prompt, samples, num_steps, guidance_scale], gallery, generate_images)
     gr.Markdown('Demo created by [Lily Berkow](https://huggingface.co/melanit/)')
 
 demo.launch()