set seed
Browse files
app.py
CHANGED
|
@@ -65,7 +65,7 @@ rmbg_model = BriaRMBG.from_pretrained("briaai/RMBG-1.4").to(device)
|
|
| 65 |
def predict(
|
| 66 |
prompt: str,
|
| 67 |
negative_prompt: str,
|
| 68 |
-
input_image: Image.Image
|
| 69 |
remove_bg: bool,
|
| 70 |
cond_mode: str,
|
| 71 |
seed: int,
|
|
@@ -192,19 +192,22 @@ def flatten(l: List[List[any]]) -> List[any]:
|
|
| 192 |
|
| 193 |
|
| 194 |
def predict_examples(
|
| 195 |
-
prompt, negative_prompt, input_image=None, remove_bg=False, cond_mode=None
|
| 196 |
):
|
|
|
|
|
|
|
|
|
|
| 197 |
return predict(
|
| 198 |
prompt,
|
| 199 |
negative_prompt,
|
| 200 |
input_image,
|
| 201 |
remove_bg,
|
| 202 |
cond_mode,
|
| 203 |
-
|
| 204 |
"euler",
|
| 205 |
"normal",
|
| 206 |
-
|
| 207 |
-
|
| 208 |
1.0,
|
| 209 |
)
|
| 210 |
|
|
@@ -256,18 +259,18 @@ with gr.Blocks(css=css) as blocks:
|
|
| 256 |
sampler_name = gr.Dropdown(
|
| 257 |
choices=samplers.KSampler.SAMPLERS,
|
| 258 |
label="Sampler Name",
|
| 259 |
-
value="
|
| 260 |
)
|
| 261 |
scheduler = gr.Dropdown(
|
| 262 |
choices=samplers.KSampler.SCHEDULERS,
|
| 263 |
label="Scheduler",
|
| 264 |
-
value="
|
| 265 |
)
|
| 266 |
steps = gr.Slider(
|
| 267 |
label="Steps", value=20, minimum=1, maximum=50, step=1
|
| 268 |
)
|
| 269 |
cfg = gr.Number(
|
| 270 |
-
label="CFG", value=
|
| 271 |
)
|
| 272 |
denoise = gr.Number(
|
| 273 |
label="Denoise", value=1.0, minimum=0.0, maximum=1.0, step=0.01
|
|
@@ -294,9 +297,9 @@ with gr.Blocks(css=css) as blocks:
|
|
| 294 |
gr.Examples(
|
| 295 |
fn=predict_examples,
|
| 296 |
examples=examples,
|
| 297 |
-
inputs=[prompt, negative_prompt, input_image, remove_bg, cond_mode],
|
| 298 |
outputs=outputs,
|
| 299 |
-
cache_examples=
|
| 300 |
)
|
| 301 |
|
| 302 |
button.click(fn=predict, inputs=inputs, outputs=outputs)
|
|
|
|
| 65 |
def predict(
|
| 66 |
prompt: str,
|
| 67 |
negative_prompt: str,
|
| 68 |
+
input_image: Image.Image,
|
| 69 |
remove_bg: bool,
|
| 70 |
cond_mode: str,
|
| 71 |
seed: int,
|
|
|
|
| 192 |
|
| 193 |
|
| 194 |
def predict_examples(
|
| 195 |
+
prompt, negative_prompt, input_image=None, remove_bg=False, cond_mode=None, seed=-1
|
| 196 |
):
|
| 197 |
+
print(
|
| 198 |
+
"RUUNING EXAMPLES", prompt, negative_prompt, input_image, remove_bg, cond_mode
|
| 199 |
+
)
|
| 200 |
return predict(
|
| 201 |
prompt,
|
| 202 |
negative_prompt,
|
| 203 |
input_image,
|
| 204 |
remove_bg,
|
| 205 |
cond_mode,
|
| 206 |
+
seed,
|
| 207 |
"euler",
|
| 208 |
"normal",
|
| 209 |
+
30,
|
| 210 |
+
10.0,
|
| 211 |
1.0,
|
| 212 |
)
|
| 213 |
|
|
|
|
| 259 |
sampler_name = gr.Dropdown(
|
| 260 |
choices=samplers.KSampler.SAMPLERS,
|
| 261 |
label="Sampler Name",
|
| 262 |
+
value="euler",
|
| 263 |
)
|
| 264 |
scheduler = gr.Dropdown(
|
| 265 |
choices=samplers.KSampler.SCHEDULERS,
|
| 266 |
label="Scheduler",
|
| 267 |
+
value="normal",
|
| 268 |
)
|
| 269 |
steps = gr.Slider(
|
| 270 |
label="Steps", value=20, minimum=1, maximum=50, step=1
|
| 271 |
)
|
| 272 |
cfg = gr.Number(
|
| 273 |
+
label="CFG", value=10.0, minimum=0.0, maximum=100.0, step=0.1
|
| 274 |
)
|
| 275 |
denoise = gr.Number(
|
| 276 |
label="Denoise", value=1.0, minimum=0.0, maximum=1.0, step=0.01
|
|
|
|
| 297 |
gr.Examples(
|
| 298 |
fn=predict_examples,
|
| 299 |
examples=examples,
|
| 300 |
+
inputs=[prompt, negative_prompt, input_image, remove_bg, cond_mode, seed],
|
| 301 |
outputs=outputs,
|
| 302 |
+
cache_examples=True,
|
| 303 |
)
|
| 304 |
|
| 305 |
button.click(fn=predict, inputs=inputs, outputs=outputs)
|
utils.py
CHANGED
|
@@ -118,6 +118,7 @@ examples = [
|
|
| 118 |
None,
|
| 119 |
False,
|
| 120 |
None,
|
|
|
|
| 121 |
],
|
| 122 |
[
|
| 123 |
"A beautiful toucan bird flying in the sky",
|
|
@@ -125,19 +126,38 @@ examples = [
|
|
| 125 |
"./examples/bg.png",
|
| 126 |
False,
|
| 127 |
"SDXL, Background",
|
|
|
|
| 128 |
],
|
| 129 |
[
|
| 130 |
-
"
|
| 131 |
"ugly distorted image, low quality, text, bad, not good ,watermark",
|
| 132 |
"./examples/lecun.png",
|
| 133 |
True,
|
| 134 |
"SDXL, Foreground",
|
|
|
|
| 135 |
],
|
| 136 |
[
|
| 137 |
-
"
|
| 138 |
"ugly distorted image, low quality, text, bad, not good ,watermark",
|
| 139 |
"./examples/julien.png",
|
| 140 |
True,
|
| 141 |
"SDXL, Foreground",
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 142 |
],
|
| 143 |
]
|
|
|
|
| 118 |
None,
|
| 119 |
False,
|
| 120 |
None,
|
| 121 |
+
1231231,
|
| 122 |
],
|
| 123 |
[
|
| 124 |
"A beautiful toucan bird flying in the sky",
|
|
|
|
| 126 |
"./examples/bg.png",
|
| 127 |
False,
|
| 128 |
"SDXL, Background",
|
| 129 |
+
123123,
|
| 130 |
],
|
| 131 |
[
|
| 132 |
+
"graphic design label of a wine bottle detail, high quality",
|
| 133 |
"ugly distorted image, low quality, text, bad, not good ,watermark",
|
| 134 |
"./examples/lecun.png",
|
| 135 |
True,
|
| 136 |
"SDXL, Foreground",
|
| 137 |
+
123123,
|
| 138 |
],
|
| 139 |
[
|
| 140 |
+
"An image of a galaxy",
|
| 141 |
"ugly distorted image, low quality, text, bad, not good ,watermark",
|
| 142 |
"./examples/julien.png",
|
| 143 |
True,
|
| 144 |
"SDXL, Foreground",
|
| 145 |
+
123123,
|
| 146 |
+
],
|
| 147 |
+
[
|
| 148 |
+
"a men jumping on swiming pool full of people",
|
| 149 |
+
"ugly distorted image, low quality, text, bad, not good ,watermark",
|
| 150 |
+
"./examples/old_jump.png",
|
| 151 |
+
False,
|
| 152 |
+
"SDXL, Foreground",
|
| 153 |
+
123123,
|
| 154 |
+
],
|
| 155 |
+
[
|
| 156 |
+
"a cute cat flying over Manhattan time square",
|
| 157 |
+
"ugly distorted image, low quality, text, bad, not good ,watermark",
|
| 158 |
+
"./examples/cat.png",
|
| 159 |
+
True,
|
| 160 |
+
"SDXL, Foreground",
|
| 161 |
+
123123,
|
| 162 |
],
|
| 163 |
]
|