Add negative prompt support

app.py

@@ -58,12 +58,13 @@ pipe.load_lora_weights("latent-consistency/lcm-lora-sdv1-5")
 pipe.fuse_lora()
 
 
-def predict(prompt, guidance, steps, seed=1231231):
+def predict(prompt, negative_prompt, guidance, steps, seed=1231231):
     generator = torch.manual_seed(seed)
     last_time = time.time()
     results = pipe(
         prompt=prompt,
         generator=generator,
+        negative_prompt=negative_prompt,
         num_inference_steps=steps,
         guidance_scale=guidance,
         width=512,
@@ -107,6 +108,9 @@ with gr.Blocks(css=css) as demo:
                 prompt = gr.Textbox(
                     placeholder="Insert your prompt here:", scale=5, container=False
                 )
+                negative_prompt = gr.Textbox(
+                    placeholder="Insert your negative prompt here:", scale=5, container=False
+                )
                 generate_bt = gr.Button("Generate", scale=1)
 
         image = gr.Image(type="filepath")
@@ -114,7 +118,7 @@ with gr.Blocks(css=css) as demo:
             guidance = gr.Slider(
                 label="Guidance", minimum=0.0, maximum=5, value=0.3, step=0.001
             )
-            steps = gr.Slider(label="Steps", value=4, minimum=2, maximum=10, step=1)
+            steps = gr.Slider(label="Steps", value=4, minimum=1, maximum=20, step=1)
             seed = gr.Slider(
                 randomize=True, minimum=0, maximum=12013012031030, label="Seed", step=1
             )