diff --git a/recipes/FaceInpainting.py b/recipes/FaceInpainting.py index 1e97a3fbe..8770740fb 100644 --- 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 index d41deb1f5..65f208a92 100644 --- 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: