Merge pull request #430 from GooeyAI/safety-checker-faceinpainting

Add safety checker for face inpainting recipe
GooeyAI · Aug 13, 2024 · 2bc7859 · 2bc7859
2 parents cb30353 + fd3ac84
commit 2bc7859
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/recipes/FaceInpainting.py b/recipes/FaceInpainting.py
@@ -252,7 +252,10 @@ def render_usage_guide(self):
     def run(self, state: dict):
         if not self.request.user.disable_safety_checker:
             yield "Running safety checker..."
-            safety_checker(image=state["input_image"])
+            safety_checker(
+                text=state["text_prompt"],
+                image=state["input_image"],
+            )
 
         yield "Extracting Face..."
 

diff --git a/recipes/Img2Img.py b/recipes/Img2Img.py
@@ -158,6 +158,7 @@ def run(self, state: dict) -> typing.Iterator[str | None]:
         yield "Generating Image..."
 
         if not self.request.user.disable_safety_checker:
+            yield "Running safety checker..."
             safety_checker(text=request.text_prompt, image=request.input_image)
 
         if request.selected_model == Img2ImgModels.instruct_pix2pix.name: