Update handler.py

handler.py

@@ -23,9 +23,9 @@ class EndpointHandler():
     def __init__(self, path=""):
         self.path = path
         # load the optimized model
-        model_id = "stabilityai/stable-diffusion-x4-upscaler"
-        #self.pipe = StableDiffusionPipeline.from_pretrained(path, torch_dtype=torch.float16)
-        #self.pipe = self.pipe.to(device)
+        #model_id = "stabilityai/stable-diffusion-x4-upscaler"
+        self.pipe = StableDiffusionPipeline.from_pretrained(path, torch_dtype=torch.float16)
+        self.pipe = self.pipe.to(device)
 
 
     def __call__(self, data) -> List[Dict[str, Any]]: