Update handler.py

handler.py

@@ -13,8 +13,9 @@ class EndpointHandler():
     
     def __init__(self, path=""):
         model_id = "timbrooks/instruct-pix2pix"
-        self.model = StableDiffusionInstructPix2PixPipeline.from_pretrained(path, torch_dtype=torch.float16, safety_checker=None).to("cuda")
-        self.model.scheduler = EulerAncestralDiscreteScheduler.from_config(self.model.scheduler.config)
+        self.pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16, safety_checker=None)
+        self.pipe.to("cuda")
+        self.pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(self.pipe.scheduler.config)
 
     def __call__(self, data):