:tada: add negative embedding

app.py

@@ -47,12 +47,13 @@ class Input:
     guidance_scale: float = 7.5
     num_inference_step: int = 28
     num_images: int = 4
+    safety_checker: bool = True
 
     def to_list(self):
         return [
             self.prompt, self.model_id, self.negative_prompt,
             self.width, self.height, self.guidance_scale,
-            self.num_inference_step, self.num_images
+            self.num_inference_step, self.num_images, self.safety_checker
         ]
 
 
@@ -85,12 +86,14 @@ def inference(
         guidance_scale: float = 7.5,
         num_inference_steps: int = 50,
         num_images: int = 4,
+        safety_checker: bool = True,
         progress=gr.Progress(track_tqdm=True),
 ) -> Image.Image:
     progress(0, "Starting inference...")
 
     global current_model_id, pipe
 
+    progress(0.1, 'Loading pipeline...')
     if model_id != current_model_id:
         try:
             # For NOT Diffusers' Models
@@ -106,12 +109,17 @@ def inference(
         except Exception as e:
             raise gr.Error(str(e))
 
+    if not safety_checker:
+        pipe.safety_checker = None
+
+    progress(0.3, 'Loading Textual Inversion...')
     # Load Textual Inversion
     pipe.load_textual_inversion(
         "checkpoints/embeddings/BadNegAnatomyV1 neg.pt", token='BadNegAnatomyV1-neg'
     )
 
     # Generation
+    progress(0.4, 'Generating images...')
     images = pipe(
         prompt,
         negative_prompt=negative_prompt,
@@ -146,8 +154,9 @@ if __name__ == "__main__":
                     value="stabilityai/stable-diffusion-3-medium-diffusers",
                 )
 
+                # Additional Input Settings
                 with gr.Accordion("Additional Settings", open=False):
-                    negative_prompt = gr.Text(label="Negative Prompt", value="")
+                    negative_prompt = gr.Text(label="Negative Prompt", value="", )
 
                     with gr.Row():
                         width = gr.Number(label="Width", value=512, step=64, minimum=64, maximum=2048)
@@ -159,6 +168,9 @@ if __name__ == "__main__":
                         label="Num Inference Steps", value=50, minimum=1, maximum=100, step=2
                     )
 
+                    with gr.Row():
+                        safety_checker = gr.Checkbox(value=True, label='Use Safety Checker')
+
             with gr.Column():
                 output_image = gr.Image(label="Image", type="pil")
 
@@ -171,6 +183,7 @@ if __name__ == "__main__":
             guidance_scale,
             num_inference_step,
             num_images,
+            safety_checker
         ]
 
         btn = gr.Button("Generate")