P01yH3dr0n commited on
Commit
e18a5b3
·
1 Parent(s): 9a80c58

character refer and inpaint fix

Browse files
Files changed (1) hide show
  1. utils.py +6 -2
utils.py CHANGED
@@ -8,7 +8,7 @@ import math
8
  import gradio as gr
9
  import hashlib
10
  import time
11
-
12
  from PIL import Image
13
 
14
  jwt_token = ''
@@ -145,7 +145,9 @@ 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
- payload['parameters']['mask'] = image2base64(inp_img['layers'][0])
 
 
149
  payload['parameters']['image'] = image2base64(inp_img['background'])
150
  payload['parameters']['extra_noise_seed'] = seed
151
  if i2i_image != None and selection == 'i2i':
@@ -158,6 +160,8 @@ def generate_novelai_image(
158
  payload['parameters']['director_reference_images'] = [image2base64(chr_image)]
159
  payload['parameters']['director_reference_descriptions'] = [{'caption': {'base_caption': 'character&style' if style_aware else 'character', 'char_captions': []}, 'legacy_uc': False}]
160
  payload['parameters']['director_reference_secondary_strength_values'] = [1 - fidelity]
 
 
161
  # Send the POST request
162
  try:
163
  response = requests.post(url, json=payload, headers=headers, timeout=180)
 
8
  import gradio as gr
9
  import hashlib
10
  import time
11
+ import PIL
12
  from PIL import Image
13
 
14
  jwt_token = ''
 
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((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)
150
+ payload['parameters']['mask'] = image2base64(mask)
151
  payload['parameters']['image'] = image2base64(inp_img['background'])
152
  payload['parameters']['extra_noise_seed'] = seed
153
  if i2i_image != None and selection == 'i2i':
 
160
  payload['parameters']['director_reference_images'] = [image2base64(chr_image)]
161
  payload['parameters']['director_reference_descriptions'] = [{'caption': {'base_caption': 'character&style' if style_aware else 'character', 'char_captions': []}, 'legacy_uc': False}]
162
  payload['parameters']['director_reference_secondary_strength_values'] = [1 - fidelity]
163
+ payload['parameters']['director_reference_strength_values'] = [1]
164
+ payload['parameters']['director_reference_information_extracted'] = [1]
165
  # Send the POST request
166
  try:
167
  response = requests.post(url, json=payload, headers=headers, timeout=180)