kadirnar commited on
Commit
0f5c23d
·
1 Parent(s): 9879d7c

Update diffusion_webui/diffusion_models/controlnet/controlnet_inpaint/controlnet_inpaint_canny.py

Browse files
diffusion_webui/diffusion_models/controlnet/controlnet_inpaint/controlnet_inpaint_canny.py CHANGED
@@ -70,9 +70,12 @@ class StableDiffusionControlNetInpaintCannyGenerator:
70
  ):
71
 
72
  normal_image = image_path["image"].convert("RGB").resize((512, 512))
73
- mask_image = image_path["mask"].convert("RGB").resize((512, 512))
74
  normal_image = np.array(normal_image)
 
 
 
75
  mask_image = np.array(mask_image)
 
76
 
77
  control_image = self.controlnet_canny_inpaint(image_path=image_path)
78
  pipe = self.load_model(
 
70
  ):
71
 
72
  normal_image = image_path["image"].convert("RGB").resize((512, 512))
 
73
  normal_image = np.array(normal_image)
74
+ normal_image = Image.fromarray(image)
75
+
76
+ mask_image = image_path["mask"].convert("RGB").resize((512, 512))
77
  mask_image = np.array(mask_image)
78
+ mask_image = Image.fromarray(image)
79
 
80
  control_image = self.controlnet_canny_inpaint(image_path=image_path)
81
  pipe = self.load_model(