add negative prompt

app.py

@@ -16,8 +16,9 @@ css = """
 }
 """
 
-def generate(prompt, turbo_steps, samp_steps, seed, progress=gr.Progress(track_tqdm=True)):
+def generate(prompt, turbo_steps, samp_steps, seed, progress=gr.Progress(track_tqdm=True), negative_prompt = ""):
     print("prompt = ", prompt)
+    print("negative prompt = ", negative_prompt)
     if seed < 0:
         seed = random.randint(1,999999)
     image = txt2img(
@@ -28,8 +29,9 @@ def generate(prompt, turbo_steps, samp_steps, seed, progress=gr.Progress(track_t
     ).images[0]
     upscaled_image = image.resize((1024,1024), 1)
     final_image = img2img(
-        prompt,
-        upscaled_image,
+        prompt=prompt,
+        negative_prompt=negative_prompt,
+        image=upscaled_image,
         num_inference_steps=samp_steps,
         guidance_scale=5,
         strength=1,
@@ -56,6 +58,7 @@ def set_base_models():
 with gr.Blocks(css=css) as demo:
     with gr.Column():
         prompt = gr.Textbox(label="Prompt")
+        negative_prompt = gr.Textbox(label="Negative Prompt")
         submit_btn = gr.Button("Generate", elem_classes="btn-green")
         
         with gr.Row():
@@ -66,7 +69,7 @@ with gr.Blocks(css=css) as demo:
             
         gallery = gr.Gallery(show_label=False, preview=True, container=False, height=1100)
         
-    submit_btn.click(generate, [prompt, turbo_steps, sampling_steps, seed], [gallery, lastSeed], queue=True)
+    submit_btn.click(generate, [prompt, turbo_steps, sampling_steps, seed, negative_prompt], [gallery, lastSeed], queue=True)
     
 txt2img, img2img = set_base_models()
-demo.launch(debug=True)
+demo.launch(debug=True)