Update app.py

app.py

@@ -129,13 +129,14 @@ class ModelManager:
         controlnet = ControlNetModel.from_single_file(
             "models/ControlNet/control_v11f1e_sd15_tile.pth", torch_dtype=torch.float16
         )
+        safety_checker = StableDiffusionSafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker")
         model_path = "models/models/Stable-diffusion/juggernaut_reborn.safetensors"
         pipe = StableDiffusionControlNetImg2ImgPipeline.from_single_file(
             model_path,
             controlnet=controlnet,
             torch_dtype=torch.float16,
             use_safetensors=True,
-            safety_checker=None
+            safety_checker=safety_checker
         )
         vae = AutoencoderKL.from_single_file(
             "models/VAE/vae-ft-mse-840000-ema-pruned.safetensors",