Update app.py

app.py

@@ -7,10 +7,12 @@ import os
 import random
 import datetime
 
-def generate_image(prompt, negative_prompt, scheduler, steps, width, height, cfg, restore_faces, seed):
+def generate_image(prompt, negative_prompt, use_negative_embedding, scheduler, steps, width, height, cfg, restore_faces, seed):
     request_time = datetime.datetime.now()
     restore_faces = bool(restore_faces)
+    use_negative_embedding = bool(use_negative_embedding)
     print(f"restore_faces: {restore_faces}, type: {type(restore_faces)}")
+    print(f"use_negative_embedding: {use_negative_embedding}, type: {type(use_negative_embedding)}")
     # Define the API endpoint
     apiUrl = os.getenv("API_URL")
     # Define the request headers
@@ -36,6 +38,12 @@ def generate_image(prompt, negative_prompt, scheduler, steps, width, height, cfg
         "width": width,
         "height": height,
         "clip_skip": 1,
+        "embeddings": [
+            {
+                "name": "rz-neg-general",
+                "strength": 1.0 if use_negative_embedding else 0
+            },
+        ],
         "vae": "vae-ft-mse-840000-ema-pruned.ckpt",
         "restore_faces": restore_faces,
         "fr_model": "CodeFormer",
@@ -87,6 +95,7 @@ iface = gr.Interface(
     inputs=[
         gr.components.Textbox(label="Prompt"),
         gr.components.Textbox(value="ugly, tiling, poorly drawn hands, poorly drawn feet, poorly drawn face, out of frame, extra limbs, disfigured, deformed, body out of frame, blurry, bad anatomy, blurred, watermark, grainy, signature, cut off, draft", label="Negative Prompt"),
+        gr.inputs.Checkbox(label="Use Negative Embedding", default=True),
         gr.components.Dropdown(choices=[
             "Euler a",
             "Euler",