Update handler.py

handler.py

@@ -15,7 +15,7 @@ if device.type != 'cuda':
 class EndpointHandler():
     def __init__(self, path=""):
         # load the optimized model
-        self.pipe = StableDiffusionPipeline.from_pretrained(path, torch_dtype=torch.float16)
+        self.pipe = StableDiffusionPipeline.from_pretrained(path, torch_dtype=torch.float16, safety_checker = None, requires_safety_checker = False)
         self.pipe = self.pipe.to(device)