Update app.py

app.py

@@ -14,10 +14,9 @@ vae_model_path = "stabilityai/sd-vae-ft-mse"
 image_encoder_path = "laion/CLIP-ViT-H-14-laion2B-s32B-b79K"
 ip_ckpt = hf_hub_download(repo_id="CiaraRowles/IP-Adapter-Instruct", filename="ip-adapter-instruct-sd15.bin", repo_type="model")
 
-#safety_model_id = "CompVis/stable-diffusion-safety-checker"
-#safety_feature_extractor = AutoFeatureExtractor.from_pretrained(safety_model_id)
-#safety_checker = StableDiffusionSafetyChecker.from_pretrained(safety_model_id)
-#the model should auto set this
+safety_model_id = "CompVis/stable-diffusion-safety-checker"
+safety_feature_extractor = AutoFeatureExtractor.from_pretrained(safety_model_id)
+safety_checker = StableDiffusionSafetyChecker.from_pretrained(safety_model_id)
 
 device = "cuda"
 
@@ -36,7 +35,8 @@ pipe = StableDiffusionPipelineCFG.from_pretrained(
     scheduler=noise_scheduler,
     vae=vae,
     torch_dtype=torch.float16,
-#    feature_extractor=safety_feature_extractor,
+    feature_extractor=safety_feature_extractor,
+    safety_checker=safety_checker
 ).to(device)
 
 #pipe.load_lora_weights("h94/IP-Adapter-FaceID", weight_name="ip-adapter-faceid-plusv2_sd15_lora.safetensors")