Update handler.py

handler.py

@@ -65,8 +65,8 @@ class EndpointHandler():
         self.controlnet = ControlNetModel.from_pretrained(CONTROLNET_MAPPING[self.control_type]["model_id"],torch_dtype=dtype).to(device)
         
         # Load StableDiffusionControlNetPipeline 
-        #self.stable_diffusion_id = "runwayml/stable-diffusion-v1-5"
-        self.stable_diffusion_id = "Lykon/dreamshaper-8"
+        self.stable_diffusion_id = "runwayml/stable-diffusion-v1-5"
+        #self.stable_diffusion_id = "Lykon/dreamshaper-8"
         self.pipe = StableDiffusionControlNetPipeline.from_pretrained(self.stable_diffusion_id, 
                                                                       controlnet=self.controlnet, 
                                                                       torch_dtype=dtype,