Update handler.py

handler.py

@@ -57,7 +57,7 @@ CONTROLNET_MAPPING = {
 class EndpointHandler():
     def __init__(self, path=""):
         # define default controlnet id and load controlnet
-        self.control_type = "normal"
+        self.control_type = "depth"
         self.controlnet = ControlNetModel.from_pretrained(CONTROLNET_MAPPING[self.control_type]["model_id"],torch_dtype=dtype).to(device)
         
         # Load StableDiffusionControlNetPipeline 
@@ -92,6 +92,7 @@ class EndpointHandler():
         
         
         # hyperparamters
+        negatice_prompt = data.pop("negative_prompt", None)
         num_inference_steps = data.pop("num_inference_steps", 30)
         guidance_scale = data.pop("guidance_scale", 7.5)
         negative_prompt = data.pop("negative_prompt", None)
@@ -126,4 +127,4 @@ class EndpointHandler():
         base64_image = base64.b64decode(image_string)
         buffer = BytesIO(base64_image)
         image = Image.open(buffer)
-        return image
+        return image