Anwar786 commited on
Commit
ed6f71c
1 Parent(s): 8861530

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +3 -2
handler.py CHANGED
@@ -57,7 +57,7 @@ CONTROLNET_MAPPING = {
57
  class EndpointHandler():
58
  def __init__(self, path=""):
59
  # define default controlnet id and load controlnet
60
- self.control_type = "normal"
61
  self.controlnet = ControlNetModel.from_pretrained(CONTROLNET_MAPPING[self.control_type]["model_id"],torch_dtype=dtype).to(device)
62
 
63
  # Load StableDiffusionControlNetPipeline
@@ -92,6 +92,7 @@ class EndpointHandler():
92
 
93
 
94
  # hyperparamters
 
95
  num_inference_steps = data.pop("num_inference_steps", 30)
96
  guidance_scale = data.pop("guidance_scale", 7.5)
97
  negative_prompt = data.pop("negative_prompt", None)
@@ -126,4 +127,4 @@ class EndpointHandler():
126
  base64_image = base64.b64decode(image_string)
127
  buffer = BytesIO(base64_image)
128
  image = Image.open(buffer)
129
- return image
 
57
  class EndpointHandler():
58
  def __init__(self, path=""):
59
  # define default controlnet id and load controlnet
60
+ self.control_type = "depth"
61
  self.controlnet = ControlNetModel.from_pretrained(CONTROLNET_MAPPING[self.control_type]["model_id"],torch_dtype=dtype).to(device)
62
 
63
  # Load StableDiffusionControlNetPipeline
 
92
 
93
 
94
  # hyperparamters
95
+ negatice_prompt = data.pop("negative_prompt", None)
96
  num_inference_steps = data.pop("num_inference_steps", 30)
97
  guidance_scale = data.pop("guidance_scale", 7.5)
98
  negative_prompt = data.pop("negative_prompt", None)
 
127
  base64_image = base64.b64decode(image_string)
128
  buffer = BytesIO(base64_image)
129
  image = Image.open(buffer)
130
+ return image