Merge pull request #152 from livepeer/vb/logging/safety_checker

fix: logs for nsfw detection

Author: emranemran

src/streamdiffusion/wrapper.py

@@ -726,6 +726,7 @@ def img2img(
                 denormalized_image_tensor = image
             if self.safety_checker(denormalized_image_tensor, self.safety_checker_threshold):
                 image = self.nsfw_fallback_img
+                logger.info(f"NSFW content detected, falling back to {self.nsfw_fallback_img} frame")
             elif self.safety_checker_fallback_type == "previous":
                 self.nsfw_fallback_img = image
 
@@ -1594,6 +1595,7 @@ def _load_model(
                             cuda_stream,
                             use_cuda_graph=True,
                         )
+                        logger.info("Safety Checker engine loaded successfully")
 
             if acceleration == "sfast":
                 from streamdiffusion.acceleration.sfast import (