Update app.py

app.py

@@ -133,13 +133,14 @@ def resize_crop(image, size=512):
 #         gr.Warning("NSFW content detected.")
 #         return Image.new("RGB", (512, 512))
 #     return results.images[0]
-async def predict(prompt, strength, steps, seed=1231231):
+async def predict(prompt, neg_prompt, strength, steps, seed=1231231):
     generator = torch.manual_seed(seed)
     last_time = time.time()
     t2i_pipe.safety_checker = None
     t2i_pipe.requires_safety_checker = False
     results = t2i_pipe(
         prompt=prompt,
+        negative_prompt = neg_prompt,
         generator=generator,
         num_inference_steps=steps,
         guidance_scale=0.0,
@@ -185,13 +186,19 @@ with gr.Blocks(css=css) as demo:
                 scale=5,
                 container=False,
             )
-            neg_prompt = gr.Textbox(
-                placeholder="Insert your negative prompt here:",
-                scale=5,
-                container=False,
-            )
+            # neg_prompt = gr.Textbox(
+            #     placeholder="Insert your negative prompt here:",
+            #     scale=5,
+            #     container=False,
+            # )
             generate_bt = gr.Button("Generate", scale=1)
         with gr.Row():
+            with gr.Column():
+                neg_prompt = gr.Textbox(
+                    placeholder="Insert your negative prompt here:",
+                    scale=5,
+                    container=False,
+                )
             # with gr.Column():
             #     image_input = gr.Image(
             #         sources=["upload", "webcam", "clipboard"],
@@ -240,10 +247,11 @@ with gr.Blocks(css=css) as demo:
         #     ```
         #     """
         #     )
-        inputs = [prompt, strength, steps, seed]
+        inputs = [prompt, neg_prompt, strength, steps, seed]
         # inputs = [image_input, prompt, strength, steps, seed]
         generate_bt.click(fn=predict, inputs=inputs, outputs=image, show_progress=False)
         prompt.input(fn=predict, inputs=inputs, outputs=image, show_progress=False)
+        neg_prompt.input(fn=predict, inputs=inputs, outputs=image, show_progress=False)
         steps.change(fn=predict, inputs=inputs, outputs=image, show_progress=False)
         seed.change(fn=predict, inputs=inputs, outputs=image, show_progress=False)
         strength.change(fn=predict, inputs=inputs, outputs=image, show_progress=False)