Add additional inputs

app.py

@@ -2,12 +2,18 @@ import gradio as gr
 import torch
 from diffusers import DiffusionPipeline
 
-def generate(prompt, negative_prompt):
+def generate(
+    prompt, negative_prompt, num_inference_steps, width, height, guidance_scale
+):
     pipeline = DiffusionPipeline.from_pretrained("Lykon/DreamShaper")
 
     return pipeline(
         prompt=prompt,
         negative_prompt=negative_prompt,
+        num_inference_steps=num_inference_steps,
+        width=width,
+        height=height,
+        guidance_scale=guidance_scale,
     ).images[0]
 
 iface = gr.Interface(
@@ -15,6 +21,10 @@ iface = gr.Interface(
     inputs=[
         gr.Textbox(label="Prompt", value=""),
         gr.Textbox(label="Negative Prompt", value=""),
+        gr.Slider(label="Sampling Steps", minimum=1, maximum=150, value=30, step=1),
+        gr.Slider(label="Width", minimum=64, maximum=2048, value=512, step=1),
+        gr.Slider(label="Height", minimum=64, maximum=2048, value=512, step=1),
+        gr.Slider(label="CFG Scale", minimum=1, maximum=30, value=9, step=0.5),
     ],
     outputs="image",
 )