Update app.py

app.py

@@ -1,6 +1,6 @@
 from diffusers import StableDiffusionXLPipeline
 import torch
-from gradio import Interface, Image
+from gradio import Interface, Image, Dropdown, Slider
 import gradio as gr
 import spaces
 
@@ -9,16 +9,23 @@ pipe = StableDiffusionXLPipeline.from_pretrained(model_id, torch_dtype=torch.flo
 pipe = pipe.to("cuda")
 
 @spaces.GPU()
-def text_to_image(prompt, progress=gr.Progress(track_tqdm=True)):
- image = pipe(prompt).images [0]
- return image
+def text_to_image(prompt, negative_prompt, steps, guidance_scale):
+    image = pipe(prompt, negative_prompt=negative_prompt, num_inference_steps=steps, guidance_scale=guidance_scale).images[0]
+    return image
 
 
 gradio_interface = Interface(
- fn=text_to_image,
- inputs="text",
- outputs=Image(type="pil", show_download_button=True),
- examples=[["person running"]],
- theme=gr.themes.Soft()
+    fn=text_to_image,
+    inputs=[
+        gr.Textbox(label="Prompt", lines=2, placeholder="Enter your prompt here..."),
+        gr.Textbox(label="Negative Prompt", lines=2, placeholder="What to exclude from the image..."),
+        gr.Slider(minimum=1, maximum=100, value=50, label="Steps", step=1),
+        gr.Slider(minimum=1, maximum=20, value=7.5, label="Guidance Scale", step=0.1)
+    ],
+    outputs=Image(type="pil", show_download_button=True),
+    examples=[
+        ["magical kitten, 4k, high quality, (masterpiece)"]
+    ],
+    theme=gr.themes.Soft()
 )
 gradio_interface.launch()