Update handler.py

handler.py

@@ -66,7 +66,9 @@ class EndpointHandler():
         self.pipe = StableDiffusionControlNetPipeline.from_pretrained(self.stable_diffusion_id, 
                                                                       controlnet=self.controlnet, 
                                                                       torch_dtype=dtype,
-                                                                      safety_checker=None).to(device)
+                                                                      #safety_checker=None).to(device)
+                                                                      safety_checker = StableDiffusionSafetyChecker.from_pretrained('CompVis/stable-diffusion-safety-checker')
+
         # Define Generator with seed
         self.generator = torch.Generator(device="cpu").manual_seed(3)