Update handler.py

handler.py

@@ -72,7 +72,8 @@ class EndpointHandler():
                                                                       torch_dtype=dtype,
                                                                       #safety_checker=None).to(device)
                                                                       #processor = AutoProcessor.from_pretrained("CompVis/stable-diffusion-safety-checker")
-                                                                      safety_checker = SafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker")
+                                                                      #safety_checker = SafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker")
+                                                                      safety_checker = StableDiffusionSafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker")
                                                    
         # Define Generator with seed
         self.generator = torch.Generator(device="cpu").manual_seed(3)