Spaces:
Running
Running
Commit
·
e18a5b3
1
Parent(s):
9a80c58
character refer and inpaint fix
Browse files
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 |
-
|
|
|
|
|
|
|
| 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)
|