oshita-n commited on
Commit
05ba6bb
·
1 Parent(s): 6291a59
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -27,8 +27,7 @@ def canny_image(image: Image.Image) -> Image.Image:
27
  return Image.fromarray(np_image)
28
 
29
  def process_image(input_image: Image.Image, to_grayscale: bool, prompt: str) -> Image.Image:
30
- output_image = remove_background(input_image, to_grayscale)
31
- canny_output = canny_image(output_image)
32
 
33
  controlnet = ControlNetModel.from_pretrained("lllyasviel/sd-controlnet-canny", torch_dtype=torch.float32)
34
  pipe = StableDiffusionControlNetPipeline.from_pretrained(
 
27
  return Image.fromarray(np_image)
28
 
29
  def process_image(input_image: Image.Image, to_grayscale: bool, prompt: str) -> Image.Image:
30
+ canny_output = canny_image(input_image)
 
31
 
32
  controlnet = ControlNetModel.from_pretrained("lllyasviel/sd-controlnet-canny", torch_dtype=torch.float32)
33
  pipe = StableDiffusionControlNetPipeline.from_pretrained(