fix incorrect run step on safety checker

GooeyAI · Aug 21, 2024 · 8f8206b · 8f8206b
1 parent 4c73a9a
commit 8f8206b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/daras_ai_v2/safety_checker.py b/daras_ai_v2/safety_checker.py
@@ -16,8 +16,8 @@ def safety_checker(*, text: str | None = None, image: str | None = None):
 
     if text:
         safety_checker_text(text)
-    # if image:
-    #     safety_checker_image(image)
+    if image:
+        safety_checker_image(image)
 
 
 def safety_checker_text(text_input: str):

diff --git a/recipes/Img2Img.py b/recipes/Img2Img.py
@@ -155,12 +155,12 @@ def run(self, state: dict) -> typing.Iterator[str | None]:
         init_image = request.input_image
         init_image_bytes = requests.get(init_image).content
 
-        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)
 
+        yield "Generating Image..."
+
         if request.selected_model == Img2ImgModels.instruct_pix2pix.name:
             state["output_images"] = instruct_pix2pix(
                 prompt=request.text_prompt,