Handle Negative prompt

app.py

@@ -5,14 +5,17 @@ from diffusers import DiffusionPipeline
 pipe = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo")
 
 @spaces.GPU(duration=250)
-def generate_image(prompt):
+def generate_image(prompt, negative_prompt):
     # Run the diffusion model to generate an image
-    output = pipe(prompt, num_inference_steps=50, guidance_scale=7.5)
+    output = pipe(prompt, negative_prompt, num_inference_steps=50, guidance_scale=7.5)
     return output.images[0]
 
+prompt = gr.Textbox(label = "Prompt", info = "Describe the subject, the background and the style of image; 77 token limit", placeholder = "Describe what you want to see", lines = 2)
+negative_prompt = gr.Textbox(label = "Negative prompt", placeholder = "Describe what you do NOT want to see", value = "Ugly, malformed, noise, blur, watermark")
+
 gr_interface = gr.Interface(
     fn=generate_image,
-    inputs="text",
+    inputs=[prompt, negative_prompt],
     outputs="image",
     title="Real-time Image Generation with Diffusion",
     description="Enter a prompt to generate an image",