P01yH3dr0n commited on
Commit
732ca9d
·
1 Parent(s): 488ed47
Files changed (1) hide show
  1. utils.py +2 -3
utils.py CHANGED
@@ -145,9 +145,8 @@ def generate_novelai_image(
145
  if selection == 'inp' and inp_img['background'] != None:
146
  payload['action'] = "infill"
147
  payload['model'] = model.replace('-preview', '') + '-inpainting'
148
- mask = inp_img['layers'][0]
149
- mask = mask.resize((width, height))
150
- mask = mask.resize((mask.size[0]//8, mask.size[1]//8), resample=PIL.Image.NEAREST).resize(mask.size, resample=PIL.Image.NEAREST).point(lambda x: 0 if x<255 else 255)
151
  payload['parameters']['mask'] = image2base64(mask)
152
  payload['parameters']['image'] = image2base64(inp_img['background'])
153
  payload['parameters']['extra_noise_seed'] = seed
 
145
  if selection == 'inp' and inp_img['background'] != None:
146
  payload['action'] = "infill"
147
  payload['model'] = model.replace('-preview', '') + '-inpainting'
148
+ mask = inp_img['layers'][0].resize((width, height))
149
+ mask = mask.resize((mask.size[0]//8, mask.size[1]//8), resample=Image.NEAREST).resize(mask.size, resample=Image.NEAREST).point(lambda x: 0 if x < 255 else 255)
 
150
  payload['parameters']['mask'] = image2base64(mask)
151
  payload['parameters']['image'] = image2base64(inp_img['background'])
152
  payload['parameters']['extra_noise_seed'] = seed