Spaces:
Paused
Paused
P01yH3dr0n
commited on
Commit
•
98c3bc5
1
Parent(s):
0c51c35
Update app.py
Browse files
app.py
CHANGED
@@ -53,13 +53,20 @@ def control_ui():
|
|
53 |
info_extract = gr.Slider(label='参考信息提取', value=1, minimum=0, maximum=1, step=0.1)
|
54 |
ref_str = gr.Slider(label='参考强度', value=0.6, minimum=0, maximum=1, step=0.1)
|
55 |
reuse_img_vibe = gr.Button(value='使用上一次生成的图片')
|
56 |
-
with gr.Accordion('
|
|
|
|
|
|
|
|
|
|
|
|
|
57 |
with gr.Row():
|
58 |
use_inp = gr.Checkbox(label='启用', value=False)
|
59 |
overlay = gr.Checkbox(label='覆盖原图', value=True)
|
60 |
-
inp_img = gr.ImageEditor(label="上传图片", value=None, interactive=True, type="pil", eraser=False, transforms=None, brush='imagemask')
|
61 |
inp_str = gr.Slider(label='重绘强度', value=0.7, minimum=0, maximum=0.99, step=0.01)
|
62 |
reuse_img_inp = gr.Button(value='使用上一次生成的图片')
|
|
|
63 |
with gr.Row():
|
64 |
with gr.Column():
|
65 |
with gr.Accordion('Advanced Gen Setting', open=False):
|
@@ -84,17 +91,17 @@ def control_ui():
|
|
84 |
rand_seed.click(fn=lambda: -1, inputs=None, outputs=seed)
|
85 |
width.change(lambda w, h: h if w*h<=1024*1024 else (1024*1024//w//64)*64, [width, height], height)
|
86 |
height.change(lambda w, h: w if w*h<=1024*1024 else (1024*1024//h//64)*64, [width, height], width)
|
87 |
-
return gen_btn,[prompt, quality_tags, neg_prompt, seed, scale, width, height, steps, sampler, scheduler, smea, dyn, dyn_threshold, cfg_rescale, ref_image, info_extract, ref_str,
|
88 |
|
89 |
|
90 |
-
def generate(prompt, quality_tags, neg_prompt, seed, scale, width, height, steps, sampler, scheduler, smea, dyn, dyn_threshold, cfg_rescale, ref_image, info_extract, ref_str,
|
91 |
global today_count
|
92 |
set_token(os.environ.get('token'))
|
93 |
img_data, payload = generate_novelai_image(
|
94 |
f"{prompt}, {quality_tags}", neg_prompt, seed, scale,
|
95 |
width, height, steps, sampler, scheduler,
|
96 |
smea, dyn, dyn_threshold, cfg_rescale, ref_image, info_extract, ref_str,
|
97 |
-
|
98 |
)
|
99 |
if not isinstance(img_data, bytes):
|
100 |
return None, payload
|
@@ -121,7 +128,7 @@ def main_ui():
|
|
121 |
gen_btn.click(generate, paras + [others[0], others[4]], [image, info])
|
122 |
others[2].click(lambda o, s: o if len(s) == 0 else s['parameters']['seed'], inputs=[paras[3], info], outputs=paras[3])
|
123 |
others[3].click(lambda i: i, inputs=image, outputs=paras[14])
|
124 |
-
others[
|
125 |
return page, paras[:14]
|
126 |
|
127 |
|
|
|
53 |
info_extract = gr.Slider(label='参考信息提取', value=1, minimum=0, maximum=1, step=0.1)
|
54 |
ref_str = gr.Slider(label='参考强度', value=0.6, minimum=0, maximum=1, step=0.1)
|
55 |
reuse_img_vibe = gr.Button(value='使用上一次生成的图片')
|
56 |
+
with gr.Accordion('图生图', open=False):
|
57 |
+
i2i_image = gr.Image(label="上传图片", value=None, sources=["upload"], interactive=True, type="pil")
|
58 |
+
i2i_str = gr.Slider(label='去噪强度', value=0.7, minimum=0, maximum=0.99, step=0.01)
|
59 |
+
i2i_noise = gr.Slider(label='噪声', value=0, minimum=0, maximum=1, step=0.1)
|
60 |
+
reuse_img_i2i = gr.Button(value='使用上一次生成的图片')
|
61 |
+
'''
|
62 |
+
with gr.Accordion('局部重绘', open=False, visible=False):
|
63 |
with gr.Row():
|
64 |
use_inp = gr.Checkbox(label='启用', value=False)
|
65 |
overlay = gr.Checkbox(label='覆盖原图', value=True)
|
66 |
+
inp_img = gr.ImageEditor(label="上传图片", value=None, sources=["upload"], interactive=True, type="pil", eraser=False, transforms=None, brush='imagemask')
|
67 |
inp_str = gr.Slider(label='重绘强度', value=0.7, minimum=0, maximum=0.99, step=0.01)
|
68 |
reuse_img_inp = gr.Button(value='使用上一次生成的图片')
|
69 |
+
'''
|
70 |
with gr.Row():
|
71 |
with gr.Column():
|
72 |
with gr.Accordion('Advanced Gen Setting', open=False):
|
|
|
91 |
rand_seed.click(fn=lambda: -1, inputs=None, outputs=seed)
|
92 |
width.change(lambda w, h: h if w*h<=1024*1024 else (1024*1024//w//64)*64, [width, height], height)
|
93 |
height.change(lambda w, h: w if w*h<=1024*1024 else (1024*1024//h//64)*64, [width, height], width)
|
94 |
+
return gen_btn,[prompt, quality_tags, neg_prompt, seed, scale, width, height, steps, sampler, scheduler, smea, dyn, dyn_threshold, cfg_rescale, ref_image, info_extract, ref_str, i2i_image, i2i_str, i2i_noise], [save, rand_seed, reuse_seed, reuse_img_vibe, reuse_img_i2i]
|
95 |
|
96 |
|
97 |
+
def generate(prompt, quality_tags, neg_prompt, seed, scale, width, height, steps, sampler, scheduler, smea, dyn, dyn_threshold, cfg_rescale, ref_image, info_extract, ref_str, i2i_image, i2i_str, i2i_noise, save):
|
98 |
global today_count
|
99 |
set_token(os.environ.get('token'))
|
100 |
img_data, payload = generate_novelai_image(
|
101 |
f"{prompt}, {quality_tags}", neg_prompt, seed, scale,
|
102 |
width, height, steps, sampler, scheduler,
|
103 |
smea, dyn, dyn_threshold, cfg_rescale, ref_image, info_extract, ref_str,
|
104 |
+
i2i_image, i2i_str, i2i_noise
|
105 |
)
|
106 |
if not isinstance(img_data, bytes):
|
107 |
return None, payload
|
|
|
128 |
gen_btn.click(generate, paras + [others[0], others[4]], [image, info])
|
129 |
others[2].click(lambda o, s: o if len(s) == 0 else s['parameters']['seed'], inputs=[paras[3], info], outputs=paras[3])
|
130 |
others[3].click(lambda i: i, inputs=image, outputs=paras[14])
|
131 |
+
others[4].click(lambda i: i, inputs=image, outputs=paras[17])
|
132 |
return page, paras[:14]
|
133 |
|
134 |
|