mattyamonaca commited on
Commit
769989d
1 Parent(s): c8d48fa
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -60,13 +60,12 @@ class webui:
60
 
61
  gen_image = generate(pipe, detectors, pos_prompt, neg_prompt)
62
  color_img, unfinished = process(gen_image.resize((image.shape[1], image.shape[0]), Image.ANTIALIAS) , org_line_image, alpha_th, thickness)
63
- color_img.save(f"{output_dir}/color_img.png")
64
-
65
  #color_img = color_img.resize((image.shape[1], image.shape[0]) , Image.ANTIALIAS)
66
 
67
 
68
  output_img = Image.alpha_composite(color_img, org_line_image)
69
  name = randomname(10)
 
70
  os.makedirs(f"{output_dir}/{name}")
71
  output_img.save(f"{output_dir}/{name}/output_image.png")
72
  org_line_image.save(f"{output_dir}/{name}/line_image.png")
 
60
 
61
  gen_image = generate(pipe, detectors, pos_prompt, neg_prompt)
62
  color_img, unfinished = process(gen_image.resize((image.shape[1], image.shape[0]), Image.ANTIALIAS) , org_line_image, alpha_th, thickness)
 
 
63
  #color_img = color_img.resize((image.shape[1], image.shape[0]) , Image.ANTIALIAS)
64
 
65
 
66
  output_img = Image.alpha_composite(color_img, org_line_image)
67
  name = randomname(10)
68
+ os.makedirs(f"{output_dir}")
69
  os.makedirs(f"{output_dir}/{name}")
70
  output_img.save(f"{output_dir}/{name}/output_image.png")
71
  org_line_image.save(f"{output_dir}/{name}/line_image.png")